Skip to content

Commit

Permalink
✏️ Fix download typo
Browse files Browse the repository at this point in the history
  • Loading branch information
Alan01252 committed Jun 5, 2024
1 parent 6777bd1 commit 65d9034
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
6 changes: 3 additions & 3 deletions docs/HOWTO.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
8 changes: 4 additions & 4 deletions yolo/tools/dataset_preparation.py
Original file line number Diff line number Diff line change
Expand Up @@ -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))
Expand Down

0 comments on commit 65d9034

Please sign in to comment.