hanjun996
5 years ago
1 changed files with 8 additions and 7 deletions
Unified View
Diff Options
+ 8
- 7
model_zoo/official/cv/ssd/src/dataset.py
View File
| @@ -233,13 +233,14 @@ def create_voc_label(is_training): | |||||
| 'id': img_id} | 'id': img_id} | ||||
| json_dict['images'].append(image) | json_dict['images'].append(image) | ||||
| for cls_name, cid in cls_map.items(): | |||||
| cat = {'supercategory': 'none', 'id': cid, 'name': cls_name} | |||||
| json_dict['categories'].append(cat) | |||||
| json_fp = open(json_file, 'w') | |||||
| json_str = json.dumps(json_dict) | |||||
| json_fp.write(json_str) | |||||
| json_fp.close() | |||||
| if not is_training: | |||||
| for cls_name, cid in cls_map.items(): | |||||
| cat = {'supercategory': 'none', 'id': cid, 'name': cls_name} | |||||
| json_dict['categories'].append(cat) | |||||
| json_fp = open(json_file, 'w') | |||||
| json_str = json.dumps(json_dict) | |||||
| json_fp.write(json_str) | |||||
| json_fp.close() | |||||
| return images, image_files_dict, image_anno_dict | return images, image_files_dict, image_anno_dict | ||||