File tree Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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 ))
You can’t perform that action at this time.
0 commit comments