Update google_utils.py (#1690)

This commit is contained in:
Glenn Jocher
2021-02-22 17:57:47 -08:00
committed by GitHub
parent d3533715ba
commit c1f8dd94b7
+5 -5
View File
@@ -21,11 +21,11 @@ def attempt_download(file, repo='ultralytics/yolov3'):
file = Path(str(file).strip().replace("'", '').lower())
if not file.exists():
try:
response = requests.get(f'https://api.github.com/repos/{repo}/releases/latest').json() # github api
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
# try:
# response = requests.get(f'https://api.github.com/repos/{repo}/releases/latest').json() # github api
# 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 = ['yolov3.pt', 'yolov3-spp.pt', 'yolov3-tiny.pt']
tag = subprocess.check_output('git tag', shell=True).decode().split()[-1]