Sum gallery dimension instead of all dimensions

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,