apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update google_utils.py (#1690)

Browse Source
This commit is contained in:
Glenn Jocher 2021-02-22 17:57:47 -08:00 committed by GitHub
parent d3533715ba
commit c1f8dd94b7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
utils/google_utils.py
View File

@ -21,13 +21,13 @@ def attempt_download(file, repo='ultralytics/yolov3'):
file = Path(str(file).strip().replace("'", '').lower()) file = Path(str(file).strip().replace("'", '').lower())
if not file.exists(): if not file.exists():
try: # try:
response = requests.get(f'https://api.github.com/repos/{repo}/releases/latest').json() # github api # 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', ...] # assets = [x['name'] for x in response['assets']] # release assets, i.e. ['yolov5s.pt', 'yolov5m.pt', ...]
tag = response['tag_name'] # i.e. 'v1.0' # tag = response['tag_name'] # i.e. 'v1.0'
except: # fallback plan # except: # fallback plan
assets = ['yolov3.pt', 'yolov3-spp.pt', 'yolov3-tiny.pt'] assets = ['yolov3.pt', 'yolov3-spp.pt', 'yolov3-tiny.pt']
tag = subprocess.check_output('git tag', shell=True).decode().split()[-1] tag = subprocess.check_output('git tag', shell=True).decode().split()[-1]
name = file.name name = file.name
if name in assets: if name in assets: