Sam Wang
commited on
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
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(
|
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
|
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()
|