glenn-jocher
commited on
Commit
•
77fd832
1
Parent(s):
8dc68fc
Update google_utils.py (#2017)
Browse files- utils/google_utils.py +2 -2
utils/google_utils.py
CHANGED
@@ -26,8 +26,8 @@ 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 = ['
|
30 |
-
tag = subprocess.check_output('git tag', shell=True).decode(
|
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 |
+
tag = subprocess.check_output('git tag', shell=True).decode().split()[-1]
|
31 |
|
32 |
name = file.name
|
33 |
if name in assets:
|