henry000 commited on
Commit
2cbea31
·
1 Parent(s): bce644c

✏️ [Fix] typo of downlaod -> download

Browse files
Files changed (1) hide show
  1. yolo/tools/dataset_preparation.py +4 -4
yolo/tools/dataset_preparation.py CHANGED
@@ -81,11 +81,11 @@ def prepare_dataset(dataset_cfg: DatasetConfig, task: str):
81
  logger.error(f"Error verifying the {dataset_type} dataset after extraction.")
82
 
83
 
84
- def prepare_weight(downlaod_link: Optional[str] = None, weight_path: str = "v9-c.pt"):
85
  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}"
89
 
90
  if not os.path.isdir(os.path.dirname(weight_path)):
91
  os.makedirs(os.path.dirname(weight_path))
 
81
  logger.error(f"Error verifying the {dataset_type} dataset after extraction.")
82
 
83
 
84
+ def prepare_weight(download_link: Optional[str] = None, weight_path: str = "v9-c.pt"):
85
  weight_name = os.path.basename(weight_path)
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}"
89
 
90
  if not os.path.isdir(os.path.dirname(weight_path)):
91
  os.makedirs(os.path.dirname(weight_path))