Change key format for `images` dict in `convert_coco` (#3419) (#3420)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pull/3423/head
Liam Parker 1 year ago committed by GitHub
parent 05760b04c4
commit 27a24b2796
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 4
      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 = []

Loading…
Cancel
Save