Fixes #232 tbh idk what the problem is but this change seems reasonab…

…le so 🤷
dbolya · Feb 25, 2020 · b5a2e23 · b5a2e23 · jasonkena · Feb 25, 2020
1 parent ca36f90
commit b5a2e23
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/data/coco.py b/data/coco.py
@@ -109,11 +109,16 @@ def pull_item(self, index):
         img_id = self.ids[index]
 
         if self.has_gt:
+            # What is this and why is this here? The targets are being loaded 2 lines down and that
+            # overwrites this, so why oh why is this here? What's the cause of this madness?
+            # But I can't risk it breaking things, so it stays. Is it mercy? No. It's fear.
             target = self.coco.imgToAnns[img_id]
+
+
             ann_ids = self.coco.getAnnIds(imgIds=img_id)
 
             # Target has {'segmentation', 'area', iscrowd', 'image_id', 'bbox', 'category_id'}
-            target = self.coco.loadAnns(ann_ids)
+            target = [x for x in self.coco.loadAnns(ann_ids) if x['image_id'] == img_id]
         else:
             target = []