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

Correct the output dir in dataloaders.py

Signed-off-by: Hongbo <12580159+ya0guang@users.noreply.github.com>
This commit is contained in:
Hongbo authored and GitHub committed 2024-03-01 12:32:29 +01:00
1 parent 574331f984
commit b939236170
1 file changed
+1 -1
+1 -1
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)