Skip to content

Commit 5277885

Browse files
NicolasHugfacebook-github-bot
authored andcommitted
[fbsync] Add default info dictionary to dataset (#9107)
Reviewed By: Dan-Flores Differential Revision: D79175060 fbshipit-source-id: 69562d2d3f3f1a6f7f8a28ec43e63622fad7419a
1 parent 29a22bf commit 5277885

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)