mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Correct the output dir in dataloaders.py (#12771)

Browse Source 
Correct the output dir in dataloaders.py

Signed-off-by: Hongbo <12580159+ya0guang@users.noreply.github.com>
This commit is contained in:
Hongbo 2024-03-01 06:32:29 -05:00 committed by GitHub
parent 574331f984
commit b939236170
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/dataloaders.py
View File

@ -1085,7 +1085,7 @@ def extract_boxes(path=DATASETS_DIR / "coco128"):
for j, x in enumerate(lb):
c = int(x[0]) # class
f = (path / "classifier") / f"{c}" / f"{path.stem}_{im_file.stem}_{j}.jpg" # new filename
f = (path / "classification") / f"{c}" / f"{path.stem}_{im_file.stem}_{j}.jpg" # new filename
if not f.parent.is_dir():
f.parent.mkdir(parents=True)