diff --git a/utils/downloads.py b/utils/downloads.py index eafa3b7ac309..998a7a582a33 100644 --- a/utils/downloads.py +++ b/utils/downloads.py @@ -60,12 +60,12 @@ def attempt_download(file, repo='ultralytics/yolov5'): # from utils.downloads i assets = [x['name'] for x in response['assets']] # release assets, i.e. ['yolov5s.pt', 'yolov5m.pt', ...] tag = response['tag_name'] # i.e. 'v1.0' except: # fallback plan - assets = ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', - 'yolov5s6.pt', 'yolov5m6.pt', 'yolov5l6.pt', 'yolov5x6.pt'] + assets = ['yolov5n.pt', 'yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', + 'yolov5n6.pt', 'yolov5s6.pt', 'yolov5m6.pt', 'yolov5l6.pt', 'yolov5x6.pt'] try: tag = subprocess.check_output('git tag', shell=True, stderr=subprocess.STDOUT).decode().split()[-1] except: - tag = 'v5.0' # current release + tag = 'v6.0' # current release if name in assets: safe_download(file,