Skip to content

Commit 342eb92

Browse files
authored
Add default info dictionary to dataset (#9107)
1 parent 6bbe010 commit 342eb92

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

references/detection/coco_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def convert_to_coco_api(ds):
123123
coco_ds = COCO()
124124
# annotation IDs need to start at 1, not 0, see torchvision issue #1530
125125
ann_id = 1
126-
dataset = {"images": [], "categories": [], "annotations": []}
126+
dataset = {"images": [], "categories": [], "annotations": [], "info": {}}
127127
categories = set()
128128
for img_idx in range(len(ds)):
129129
# find better way to get target

0 commit comments

Comments
 (0)