aboutsummaryrefslogtreecommitdiff
path: root/evaluate.py
diff options
context:
space:
mode:
authornunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal>2018-12-10 14:55:20 +0000
committernunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal>2018-12-10 14:55:20 +0000
commitcc0ce36fb75f4b207311c07daf86b835aea0a745 (patch)
tree87e309ab631c7a6e8abf3ecbc6e9917d89f2168c /evaluate.py
parentbe248eb34f8eec8f5436d910075695e5cf0b867c (diff)
downloadvz215_np1915-cc0ce36fb75f4b207311c07daf86b835aea0a745.tar.gz
vz215_np1915-cc0ce36fb75f4b207311c07daf86b835aea0a745.tar.bz2
vz215_np1915-cc0ce36fb75f4b207311c07daf86b835aea0a745.zip
Fix mAP
Diffstat (limited to 'evaluate.py')
-rwxr-xr-xevaluate.py3
1 files changed, 2 insertions, 1 deletions
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