Skip to content

Commit 5a595e6

Browse files
committed
✏️ Fix download typo
1 parent 7dd7b62 commit 5a595e6

2 files changed

Lines changed: 7 additions & 7 deletions

File tree

docs/HOWTO.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -138,10 +138,10 @@ Custom transformations should be designed to accept an image and its bounding bo
138138
- `func` draw_bboxes: given a image and list of bbox, draw bbox on the image
139139
- `func` draw_model: visualize the given model
140140
- **get_dataset**
141-
- `func` download_file: for a given link, downlaod the file
142-
- `func` unzip_file: unzip the downlaoded zip to data/
141+
- `func` download_file: for a given link, download the file
142+
- `func` unzip_file: unzip the downloaded zip to data/
143143
- `func` check_files: check if the dataset file numbers is correct
144-
- `func` prepare_dataset: automatic downlaod the dataset and check if it is correct
144+
- `func` prepare_dataset: automatic download the dataset and check if it is correct
145145
- **loss**
146146
- `class` BoxLoss: a Custom Loss for bounding box
147147
- `class` YOLOLoss: a implementation of yolov9 loss

yolo/tools/dataset_preparation.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -81,10 +81,10 @@ def prepare_dataset(cfg: DatasetConfig):
8181
logger.error(f"Error verifying the {dataset_type} dataset after extraction.")
8282

8383

84-
def prepare_weight(downlaod_link: Optional[str] = None, weight_name: str = "v9-c.pt"):
85-
if downlaod_link is None:
86-
downlaod_link = "https://github.com/WongKinYiu/yolov9mit/releases/download/v1.0-alpha/"
87-
weight_link = f"{downlaod_link}{weight_name}"
84+
def prepare_weight(download_link: Optional[str] = None, weight_name: str = "v9-c.pt"):
85+
if download_link is None:
86+
download_link = "https://github.com/WongKinYiu/yolov9mit/releases/download/v1.0-alpha/"
87+
weight_link = f"{download_link}{weight_name}"
8888

8989
if os.path.exists(weight_name):
9090
logger.info(f"Weight file '{weight_name}' already exists.")

0 commit comments

Comments
 (0)