| import os | |
| import json | |
| OUTPUT_PATH = './output.json' | |
| result = [] | |
| with open(OUTPUT_PATH, 'r') as f: | |
| output = json.load(f) | |
| for img_id, objects in output.items(): | |
| print(img_id) | |
| for obj in objects: | |
| result.append({ | |
| 'image_id': int(img_id), | |
| 'category_id': obj['name'], | |
| 'bbox': [obj['bbox'][0], obj['bbox'][1], obj['bbox'][2], obj['bbox'][3]], | |
| 'score': obj['confidence'] | |
| }) | |
| with open('coco_instances_results.json', 'w') as f: | |
| json.dump(result, f) | |