Sam Wang commited on
Commit
24b85bd
·
unverified ·
1 Parent(s): b18186e

Fix a typo and correct the directory path when auto downloading dataset (#20)

Browse files

* 🐛 [Fix] path error when preparing dataset

* ✏️ [Update] import typos in dataset_preparation.py

Files changed (1) hide show
  1. yolo/tools/dataset_preparation.py +2 -2
yolo/tools/dataset_preparation.py CHANGED
@@ -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)
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 @@ if __name__ == "__main__":
78
  import sys
79
 
80
  sys.path.append("./")
81
- from tools.log_helper import custom_logger
82
 
83
  custom_logger()
84
  prepare_dataset()
 
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(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
  import sys
79
 
80
  sys.path.append("./")
81
+ from utils.logging_utils import custom_logger
82
 
83
  custom_logger()
84
  prepare_dataset()