We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b18186e commit 24b85bdCopy full SHA for 24b85bd
yolo/tools/dataset_preparation.py
@@ -61,7 +61,7 @@ def prepare_dataset(cfg):
61
extract_to = os.path.join(data_dir, data_type) if data_type != "annotations" else data_dir
62
final_place = os.path.join(extract_to, dataset_type)
63
64
- os.makedirs(extract_to, exist_ok=True)
+ os.makedirs(final_place, exist_ok=True)
65
if check_files(final_place, dataset_args.get("file_num")):
66
logger.info(f"✅ Dataset {dataset_type: <12} already verified.")
67
continue
@@ -78,7 +78,7 @@ def prepare_dataset(cfg):
78
import sys
79
80
sys.path.append("./")
81
- from tools.log_helper import custom_logger
+ from utils.logging_utils import custom_logger
82
83
custom_logger()
84
prepare_dataset()
0 commit comments