xdobro23
/
YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
single_channel
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 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

4
ultralytics/yolo/data/converter.py
Unescape View File

@ -55,7 +55,7 @@ def convert_coco(labels_dir='../coco/annotations/', use_segments=False, use_keyp
data = json.load(f) data = json.load(f)
# Create image dict # 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 # Create image-annotations dict
imgToAnns = defaultdict(list) imgToAnns = defaultdict(list)
for ann in data['annotations']: for ann in data['annotations']:
@ -63,7 +63,7 @@ def convert_coco(labels_dir='../coco/annotations/', use_segments=False, use_keyp
# Write labels file # Write labels file
for img_id, anns in tqdm(imgToAnns.items(), desc=f'Annotations {json_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'] h, w, f = img['height'], img['width'], img['file_name']
bboxes = [] bboxes = []

Write Preview
Loading…
Cancel
Save