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>
ultralytics · Mar 1, 2024 · b939236 · b939236
1 parent 574331f
commit b939236
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/dataloaders.py b/utils/dataloaders.py
@@ -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)