From c053a23b2fa24269270be346977048b73f69b236 Mon Sep 17 00:00:00 2001 From: nunzip Date: Tue, 11 Dec 2018 19:26:29 +0000 Subject: Revert mAP to interpolate --- opt.py | 4 ++-- 1 file 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) -- cgit v1.2.3-54-g00ecf