From ec7d85a066592bfbed2f6c4c56f9f0b6269897e3 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Sat, 13 Feb 2021 19:41:47 +0800 Subject: Sum gallery dimension instead of all dimensions --- eval.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'eval.py') 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, -- cgit v1.2.3