From 27a24b2796caab8758bfbb2dee2417b1efe1e39d Mon Sep 17 00:00:00 2001 From: Liam Parker <17858650+GreatGameDota@users.noreply.github.com> Date: Tue, 27 Jun 2023 14:01:26 -0500 Subject: [PATCH] Change key format for `images` dict in `convert_coco` (#3419) (#3420) Co-authored-by: Glenn Jocher --- ultralytics/yolo/data/converter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 = []