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)