Fixing issue #1530 by starting ann_id to 1 in convert_to_coco_api (#1531)

13b35ffa · fsavard-eai · Francisco Massa · 7d509c5d · 13b35ffa
Commit 13b35ffa authored Oct 29, 2019 by fsavard-eai Committed by Francisco Massa Oct 29, 2019
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

references/detection/coco_utils.py references/detection/coco_utils.py +2 -1

No files found.
--- a/references/detection/coco_utils.py
+++ b/references/detection/coco_utils.py
@@ -145,7 +145,8 @@ def _coco_remove_images_without_annotations(dataset, cat_list=None):

 def convert_to_coco_api(ds):
    coco_ds = COCO()
-    ann_id = 0
+    # annotation IDs need to start at 1, not 0, see torchvision issue #1530
+    ann_id = 1
    dataset = {'images': [], 'categories': [], 'annotations': []}
    categories = set()
    for img_idx in range(len(ds)):