fix bug

dongan-beta · Aug 29, 2019 · acd5495 · acd5495
1 parent 50cbe41
commit acd5495
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dirtorch/datasets/generic.py b/dirtorch/datasets/generic.py
@@ -206,7 +206,7 @@ def eval_query_AP(self, query_idx, scores):
             gt, scores = gt[keep], scores[keep]
             gt_sorted = gt[np.argsort(scores)[::-1]]
             positive_rank = np.where(gt_sorted == 1)[0]
-            ap = compute_average_precision(positive_rank)
+            return compute_average_precision(positive_rank)
         else:
             d = {}
             for mode in ('easy', 'medium', 'hard'):