aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xevaluate.py5
1 files changed, 2 insertions, 3 deletions
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)