@@ -62,7 +62,7 @@ def create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbos | |||
def custom(path='path/to/model.pt', autoshape=True, verbose=True): | |||
# YOLOv5 custom or local model | |||
return create(path, autoshape, verbose) | |||
return create(path, autoshape=autoshape, verbose=verbose) | |||
def yolov5s(pretrained=True, channels=3, classes=80, autoshape=True, verbose=True): |
@@ -21,6 +21,7 @@ def attempt_download(file, repo='ultralytics/yolov5'): | |||
file = Path(str(file).strip().replace("'", '')) | |||
if not file.exists(): | |||
file.parent.mkdir(parents=True, exist_ok=True) # make parent dir (if required) | |||
try: | |||
response = requests.get(f'https://api.github.com/repos/{repo}/releases/latest').json() # github api | |||
assets = [x['name'] for x in response['assets']] # release assets, i.e. ['yolov5s.pt', 'yolov5m.pt', ...] |