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