diff --git a/docs/HOWTO.md b/docs/HOWTO.md index f322ba9e..75eb9260 100644 --- a/docs/HOWTO.md +++ b/docs/HOWTO.md @@ -138,10 +138,10 @@ Custom transformations should be designed to accept an image and its bounding bo - `func` draw_bboxes: given a image and list of bbox, draw bbox on the image - `func` draw_model: visualize the given model - **get_dataset** - - `func` download_file: for a given link, downlaod the file - - `func` unzip_file: unzip the downlaoded zip to data/ + - `func` download_file: for a given link, download the file + - `func` unzip_file: unzip the downloaded zip to data/ - `func` check_files: check if the dataset file numbers is correct - - `func` prepare_dataset: automatic downlaod the dataset and check if it is correct + - `func` prepare_dataset: automatic download the dataset and check if it is correct - **loss** - `class` BoxLoss: a Custom Loss for bounding box - `class` YOLOLoss: a implementation of yolov9 loss diff --git a/yolo/tools/dataset_preparation.py b/yolo/tools/dataset_preparation.py index 35c79cb7..72c529de 100644 --- a/yolo/tools/dataset_preparation.py +++ b/yolo/tools/dataset_preparation.py @@ -81,11 +81,11 @@ def prepare_dataset(cfg: DatasetConfig): logger.error(f"Error verifying the {dataset_type} dataset after extraction.") -def prepare_weight(downlaod_link: Optional[str] = None, weight_path: str = "v9-c.pt"): +def prepare_weight(download_link: Optional[str] = None, weight_path: str = "v9-c.pt"): weight_name = os.path.basename(weight_path) - if downlaod_link is None: - downlaod_link = "https://github.com/WongKinYiu/yolov9mit/releases/download/v1.0-alpha/" - weight_link = f"{downlaod_link}{weight_name}" + if download_link is None: + download_link = "https://github.com/WongKinYiu/yolov9mit/releases/download/v1.0-alpha/" + weight_link = f"{download_link}{weight_name}" if not os.path.isdir(os.path.dirname(weight_path)): os.makedirs(os.path.dirname(weight_path))