diff --git a/references/detection/coco_utils.py b/references/detection/coco_utils.py index efbbdd3f4ea..26701a2cbee 100644 --- a/references/detection/coco_utils.py +++ b/references/detection/coco_utils.py @@ -145,7 +145,8 @@ def _has_valid_annotation(anno): 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)):