Skip to content

Commit 65d9034

Browse files
committed
✏️ Fix download typo
1 parent 6777bd1 commit 65d9034

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

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,11 +81,11 @@ 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_path: str = "v9-c.pt"):
84+
def prepare_weight(download_link: Optional[str] = None, weight_path: str = "v9-c.pt"):
8585
weight_name = os.path.basename(weight_path)
86-
if downlaod_link is None:
87-
downlaod_link = "https://github.com/WongKinYiu/yolov9mit/releases/download/v1.0-alpha/"
88-
weight_link = f"{downlaod_link}{weight_name}"
86+
if download_link is None:
87+
download_link = "https://github.com/WongKinYiu/yolov9mit/releases/download/v1.0-alpha/"
88+
weight_link = f"{download_link}{weight_name}"
8989

9090
if not os.path.isdir(os.path.dirname(weight_path)):
9191
os.makedirs(os.path.dirname(weight_path))

0 commit comments

Comments
 (0)