diff --git a/ultralytics/yolo/data/converter.py b/ultralytics/yolo/data/converter.py index aa391b6..c1278dd 100644 --- a/ultralytics/yolo/data/converter.py +++ b/ultralytics/yolo/data/converter.py @@ -55,7 +55,7 @@ def convert_coco(labels_dir='../coco/annotations/', use_segments=False, use_keyp data = json.load(f) # Create image dict - images = {'%g' % x['id']: x for x in data['images']} + images = {f'{x["id"]:d}': x for x in data['images']} # Create image-annotations dict imgToAnns = defaultdict(list) for ann in data['annotations']: @@ -63,7 +63,7 @@ def convert_coco(labels_dir='../coco/annotations/', use_segments=False, use_keyp # Write labels file for img_id, anns in tqdm(imgToAnns.items(), desc=f'Annotations {json_file}'): - img = images['%g' % img_id] + img = images[f'{img_id:d}'] h, w, f = img['height'], img['width'], img['file_name'] bboxes = []