From cc0ce36fb75f4b207311c07daf86b835aea0a745 Mon Sep 17 00:00:00 2001
From: nunzip
 <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal>
Date: Mon, 10 Dec 2018 14:55:20 +0000
Subject: Fix mAP

---
 evaluate.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/evaluate.py b/evaluate.py
index 5948acc..251e952 100755
--- a/evaluate.py
+++ b/evaluate.py
@@ -159,7 +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(max_level_precision[i])/11
+                mAP[i] = sum(precision[i])/args.neighbors
             print('mAP:',np.mean(mAP))
                     
     return target_pred
-- 
cgit v1.2.3-54-g00ecf