aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xopt.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/opt.py b/opt.py
index 3d68adc..4fe263a 100755
--- a/opt.py
+++ b/opt.py
@@ -159,8 +159,8 @@ def test_model(gallery_data, probe_data, gallery_label, probe_label, gallery_cam
max_level_precision[i][j] = np.max(precision[i][np.where(recall[i]>=(j/10))])
#print(mAP[i])
for i in range(probe_label.shape[0]):
- #mAP[i] = sum(max_level_precision[i])/11
- mAP[i] = sum(precision[i])/args.neighbors
+ mAP[i] = sum(max_level_precision[i])/11
+ #mAP[i] = sum(precision[i])/args.neighbors
print('mAP:',np.mean(mAP))
return np.mean(mAP)