Browse Source

Hubconf.py bug fix (#3007)

modifyDataloader
Glenn Jocher GitHub 3 years ago
parent
commit
31ee54c28c
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions
  1. +1
    -1
      hubconf.py
  2. +1
    -0
      utils/google_utils.py

+ 1
- 1
hubconf.py View File

@@ -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):

+ 1
- 0
utils/google_utils.py View File

@@ -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', ...]

Loading…
Cancel
Save