aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authornunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal>2018-12-04 20:02:08 +0000
committernunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal>2018-12-04 20:02:08 +0000
commit390bc568d1b453da960569b26b361c338cf22e2c (patch)
tree9beb6d89d335aea26cc37301c5b7b966a09b4800
parent083b5778fe1940b23389a8457f322eaef13a3e98 (diff)
downloadvz215_np1915-390bc568d1b453da960569b26b361c338cf22e2c.tar.gz
vz215_np1915-390bc568d1b453da960569b26b361c338cf22e2c.tar.bz2
vz215_np1915-390bc568d1b453da960569b26b361c338cf22e2c.zip
Fix Bug on -n flag
-rwxr-xr-xpart2.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/part2.py b/part2.py
index 1d6485c..9a30aad 100755
--- a/part2.py
+++ b/part2.py
@@ -97,8 +97,10 @@ def test_model(gallery_data, probe_data, gallery_label, probe_label, gallery_cam
nneighbors = np.zeros((ranklist.shape[0],nsize))
nnshowrank = (np.zeros((ranklist.shape[0],nsize))).astype(object)
+
for i in range(args.multrank):
- args.neighbors = test_table[i]
+ if args.multrank!= 1:
+ args.neighbors = test_table[i]
for probe_idx in range(probe_data.shape[0]):
row = ranklist[probe_idx]
n = 0