glenn-jocher
commited on
Commit
•
14d2d2d
1
Parent(s):
cdb678f
Update google_utils.py (#2900)
Browse files- utils/google_utils.py +6 -2
utils/google_utils.py
CHANGED
@@ -26,8 +26,12 @@ def attempt_download(file, repo='ultralytics/yolov5'):
|
|
26 |
assets = [x['name'] for x in response['assets']] # release assets, i.e. ['yolov5s.pt', 'yolov5m.pt', ...]
|
27 |
tag = response['tag_name'] # i.e. 'v1.0'
|
28 |
except: # fallback plan
|
29 |
-
assets = ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt'
|
30 |
-
|
|
|
|
|
|
|
|
|
31 |
|
32 |
name = file.name
|
33 |
if name in assets:
|
|
|
26 |
assets = [x['name'] for x in response['assets']] # release assets, i.e. ['yolov5s.pt', 'yolov5m.pt', ...]
|
27 |
tag = response['tag_name'] # i.e. 'v1.0'
|
28 |
except: # fallback plan
|
29 |
+
assets = ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt',
|
30 |
+
'yolov5s6.pt', 'yolov5m6.pt', 'yolov5l6.pt', 'yolov5x6.pt']
|
31 |
+
try:
|
32 |
+
tag = subprocess.check_output('git tag', shell=True, stderr=subprocess.STDOUT).decode().split()[-1]
|
33 |
+
except:
|
34 |
+
tag = 'v5.0' # current release
|
35 |
|
36 |
name = file.name
|
37 |
if name in assets:
|