Dataset autodownload fstring updates

This commit is contained in:
Glenn Jocher committed 2022-05-31 12:36:20 +02:00
1 parent 258f4c6438
commit 487edd5609
2 files changed
+5 -5

No files matched your search

+1 -1
View File
@@ -32,7 +32,7 @@ download: |
for annot in tqdm(a['annotations'], desc=f"Converting {set} to YOLOv5 format..."):
img_id = annot['image_id']
img_name = a['images'][img_id]['name']
img_label_name = img_name[:-3] + "txt"
img_label_name = f'{img_name[:-3]}txt'
cls = annot['category_id'] # instance class id
x_center, y_center, width, height = annot['bbox']