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()
|