summaryrefslogtreecommitdiff
path: root/eval.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-14 20:50:34 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-14 20:50:34 +0800
commit10944fda51563b66cf441747f7a1b292236096cf (patch)
tree88c7ac4d1e23d39b6f19c79b701c628a3e304361 /eval.py
parent156fb6d957efda8b897c172d70eccc0d2016b2bf (diff)
parent34d2f9017e77a7bdef761ab3d92cd0340c5154c3 (diff)
Merge branch 'python3.8' into python3.7
Diffstat (limited to 'eval.py')
-rw-r--r--eval.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/eval.py b/eval.py
index ec48066..c0505f8 100644
--- a/eval.py
+++ b/eval.py
@@ -22,7 +22,7 @@ for n in range(rank):
print(f'===Rank-{n + 1} Accuracy===')
for (condition, accuracy_c) in accuracy.items():
acc_excl_identical_view = accuracy_c[:, :, n].fill_diagonal_(0)
- num_gallery_views = (acc_excl_identical_view != 0).sum()
+ num_gallery_views = (acc_excl_identical_view != 0).sum(0)
acc_each_angle = acc_excl_identical_view.sum(0) / num_gallery_views
print('{0}: {1} mean: {2:5.2f}'.format(
condition, acc_each_angle.cpu().numpy() * 100,