[Fix] fix eval_recalls error in voc_metric (#10770)

open-mmlab · Aug 14, 2023 · 635c4bf · 635c4bf
1 parent a066ef4
commit 635c4bf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mmdet/evaluation/metrics/voc_metric.py b/mmdet/evaluation/metrics/voc_metric.py
@@ -157,11 +157,11 @@ def compute_metrics(self, results: list) -> dict:
             eval_results['mAP'] = sum(mean_aps) / len(mean_aps)
             eval_results.move_to_end('mAP', last=False)
         elif self.metric == 'recall':
-            # TODO: Currently not checked.
-            gt_bboxes = [ann['bboxes'] for ann in self.annotations]
+            gt_bboxes = [gt['bboxes'] for gt in gts]
+            pr_bboxes = [pred[0] for pred in preds]
             recalls = eval_recalls(
                 gt_bboxes,
-                results,
+                pr_bboxes,
                 self.proposal_nums,
                 self.iou_thrs,
                 logger=logger,