Skip to content

Commit 24b85bd

Browse files
authored
Fix a typo and correct the directory path when auto downloading dataset (#20)
* 🐛 [Fix] path error when preparing dataset * ✏️ [Update] import typos in dataset_preparation.py
1 parent b18186e commit 24b85bd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

yolo/tools/dataset_preparation.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def prepare_dataset(cfg):
6161
extract_to = os.path.join(data_dir, data_type) if data_type != "annotations" else data_dir
6262
final_place = os.path.join(extract_to, dataset_type)
6363

64-
os.makedirs(extract_to, exist_ok=True)
64+
os.makedirs(final_place, exist_ok=True)
6565
if check_files(final_place, dataset_args.get("file_num")):
6666
logger.info(f"✅ Dataset {dataset_type: <12} already verified.")
6767
continue
@@ -78,7 +78,7 @@ def prepare_dataset(cfg):
7878
import sys
7979

8080
sys.path.append("./")
81-
from tools.log_helper import custom_logger
81+
from utils.logging_utils import custom_logger
8282

8383
custom_logger()
8484
prepare_dataset()

0 commit comments

Comments
 (0)