glenn-jocher commited on
Commit
14d2d2d
1 Parent(s): cdb678f

Update google_utils.py (#2900)

Browse files
Files changed (1) hide show
  1. 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
- tag = subprocess.check_output('git tag', shell=True).decode().split()[-1]
 
 
 
 
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: