diff options
-rwxr-xr-x | evaluate.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/evaluate.py b/evaluate.py index 7a5e3ca..5948acc 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) |