Skip to content

Commit

Permalink
Fix a typo and correct the directory path when auto downloading datas…
Browse files Browse the repository at this point in the history
…et (#20)

* 🐛 [Fix] path error when preparing dataset

* ✏️ [Update] import typos in dataset_preparation.py
  • Loading branch information
SamWang0807 authored Jun 3, 2024
1 parent b18186e commit 24b85bd
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions yolo/tools/dataset_preparation.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ def prepare_dataset(cfg):
extract_to = os.path.join(data_dir, data_type) if data_type != "annotations" else data_dir
final_place = os.path.join(extract_to, dataset_type)

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

sys.path.append("./")
from tools.log_helper import custom_logger
from utils.logging_utils import custom_logger

custom_logger()
prepare_dataset()

0 comments on commit 24b85bd

Please sign in to comment.