diff --git a/references/detection/coco_utils.py b/references/detection/coco_utils.py index f40dcdff783..44b917a6ec6 100644 --- a/references/detection/coco_utils.py +++ b/references/detection/coco_utils.py @@ -123,7 +123,7 @@ def convert_to_coco_api(ds): coco_ds = COCO() # annotation IDs need to start at 1, not 0, see torchvision issue #1530 ann_id = 1 - dataset = {"images": [], "categories": [], "annotations": []} + dataset = {"images": [], "categories": [], "annotations": [], "info": {}} categories = set() for img_idx in range(len(ds)): # find better way to get target