diff options
author | nunzip <np.scarh@gmail.com> | 2018-12-13 00:53:43 +0000 |
---|---|---|
committer | nunzip <np.scarh@gmail.com> | 2018-12-13 00:53:43 +0000 |
commit | 34ef39354a48146fff99d9fcbb1882ae50f9a627 (patch) | |
tree | 456e199100c81e3201676ff6383664590fd60fb1 /evaluate.py | |
parent | 9631be5a9b9e90f74b3484632ab5d9f379334a50 (diff) | |
download | vz215_np1915-34ef39354a48146fff99d9fcbb1882ae50f9a627.tar.gz vz215_np1915-34ef39354a48146fff99d9fcbb1882ae50f9a627.tar.bz2 vz215_np1915-34ef39354a48146fff99d9fcbb1882ae50f9a627.zip |
Fix evaluate -M -A bug
Diffstat (limited to 'evaluate.py')
-rwxr-xr-x | evaluate.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/evaluate.py b/evaluate.py index b178abc..a19a7a9 100755 --- a/evaluate.py +++ b/evaluate.py @@ -155,8 +155,11 @@ def test_model(gallery_data, probe_data, gallery_label, probe_label, gallery_cam AP[i] = sum(max_level_precision[i])/11 mAP = np.mean(AP) print('mAP:',mAP) - return target_pred, mAP - return target_pred + + if args.mAP: + return target_pred, mAP + else: + return target_pred def main(): logging.debug("Verbose mode is on") |