Merge pull request #36 from FredericOdermatt/fix_category_agnostic

(fix) return non empty lists for category agnostic evaluation
lvis-dataset · Feb 21, 2023 · da5f65d · da5f65d
2 parents 35f09cd + a1fa536
commit da5f65d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lvis/eval.py b/lvis/eval.py
@@ -155,12 +155,12 @@ def _get_gt_dt(self, img_id, cat_id):
             gt = [
                 _ann
                 for _cat_id in self.params.cat_ids
-                for _ann in self._gts[img_id, cat_id]
+                for _ann in self._gts[img_id, _cat_id]
             ]
             dt = [
                 _ann
                 for _cat_id in self.params.cat_ids
-                for _ann in self._dts[img_id, cat_id]
+                for _ann in self._dts[img_id, _cat_id]
             ]
         return gt, dt