in download_imgur5k.py [0:0]
def _create_split_json(anno_json, _split_idx):
split_json = {}
split_json['index_id'] = {}
split_json['index_to_ann_map'] = {}
split_json['ann_id'] = {}
for _idx in _split_idx:
# Check if the idx is not bad
if _idx not in anno_json['index_id']:
continue
split_json['index_id'][_idx] = anno_json['index_id'][_idx]
split_json['index_to_ann_map'][_idx] = anno_json['index_to_ann_map'][_idx]
for ann_id in split_json['index_to_ann_map'][_idx]:
split_json['ann_id'][ann_id] = anno_json['ann_id'][ann_id]
return split_json