diff options
| -rwxr-xr-x | evaluate.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/evaluate.py b/evaluate.py index 642116f..696356e 100755 --- a/evaluate.py +++ b/evaluate.py @@ -158,8 +158,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 target_pred |