Fix bug in creating img_nl in eval

lvis-dataset · Aug 12, 2019 · 73142a4 · 73142a4
1 parent 2130954
commit 73142a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lvis/eval.py b/lvis/eval.py
@@ -87,7 +87,7 @@ def _prepare(self):
         # presence or absence in an image.
         img_data = self.lvis_gt.load_imgs(ids=self.params.img_ids)
         # per image map of categories not present in image
-        img_nl = {d["id"]: d["not_exhaustive_category_ids"] for d in img_data}
+        img_nl = {d["id"]: d["neg_category_ids"] for d in img_data}
         # per image list of categories present in image
         img_pl = defaultdict(set)
         for ann in gts: