From 743ac705dac81896f30227fb949c3175ef751427 Mon Sep 17 00:00:00 2001 From: nunzip Date: Mon, 10 Dec 2018 02:42:54 +0000 Subject: Spaces --- evaluate.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'evaluate.py') diff --git a/evaluate.py b/evaluate.py index 9218649..9910360 100755 --- a/evaluate.py +++ b/evaluate.py @@ -56,9 +56,8 @@ parser.add_argument("-V", "--validation", help="Validation Mode", action='store_ parser.add_argument("-K", "--kmean_alt", help="Kmean", type=int, default=0) parser.add_argument("-P", "--mAP", help="Mean Average Precision", action='store_true') - - args = parser.parse_args() + if args.verbose: logging.basicConfig(level=logging.DEBUG) @@ -223,7 +222,7 @@ def main(): if(args.kmean): debug("Using Kmeans") train_data, train_label, train_cam = create_kmean_clusters(feature_vectors, labels,gallery_idx,camId) - + if args.kmean_alt: kmeans = KMeans(n_clusters=args.kmean_alt, random_state=0).fit(train_data) neigh = NearestNeighbors(n_neighbors=1) -- cgit v1.2.3