aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-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