aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authornunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal>2018-12-10 02:42:54 +0000
committernunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal>2018-12-10 02:42:54 +0000
commit743ac705dac81896f30227fb949c3175ef751427 (patch)
treead3664a1042a5156bee3b7336c788e85e8de3818
parenta1985d4f28fdf596f4c223b34598f9d82eb5a154 (diff)
downloadvz215_np1915-743ac705dac81896f30227fb949c3175ef751427.tar.gz
vz215_np1915-743ac705dac81896f30227fb949c3175ef751427.tar.bz2
vz215_np1915-743ac705dac81896f30227fb949c3175ef751427.zip
Spaces
-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)