Commit
•
e520e82
1
Parent(s):
d0bfeb3
Update autodownload fallbacks to v6.0 assets (#5177)
Browse files- utils/downloads.py +3 -3
utils/downloads.py
CHANGED
|
@@ -60,12 +60,12 @@ def attempt_download(file, repo='ultralytics/yolov5'): # from utils.downloads i
|
|
| 60 |
assets = [x['name'] for x in response['assets']] # release assets, i.e. ['yolov5s.pt', 'yolov5m.pt', ...]
|
| 61 |
tag = response['tag_name'] # i.e. 'v1.0'
|
| 62 |
except: # fallback plan
|
| 63 |
-
assets = ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt',
|
| 64 |
-
'yolov5s6.pt', 'yolov5m6.pt', 'yolov5l6.pt', 'yolov5x6.pt']
|
| 65 |
try:
|
| 66 |
tag = subprocess.check_output('git tag', shell=True, stderr=subprocess.STDOUT).decode().split()[-1]
|
| 67 |
except:
|
| 68 |
-
tag = '
|
| 69 |
|
| 70 |
if name in assets:
|
| 71 |
safe_download(file,
|
|
|
|
| 60 |
assets = [x['name'] for x in response['assets']] # release assets, i.e. ['yolov5s.pt', 'yolov5m.pt', ...]
|
| 61 |
tag = response['tag_name'] # i.e. 'v1.0'
|
| 62 |
except: # fallback plan
|
| 63 |
+
assets = ['yolov5n.pt', 'yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt',
|
| 64 |
+
'yolov5n6.pt', 'yolov5s6.pt', 'yolov5m6.pt', 'yolov5l6.pt', 'yolov5x6.pt']
|
| 65 |
try:
|
| 66 |
tag = subprocess.check_output('git tag', shell=True, stderr=subprocess.STDOUT).decode().split()[-1]
|
| 67 |
except:
|
| 68 |
+
tag = 'v6.0' # current release
|
| 69 |
|
| 70 |
if name in assets:
|
| 71 |
safe_download(file,
|