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

Update google_utils.py

Browse Source
This commit is contained in:
Glenn Jocher 2021-02-16 11:09:54 -08:00 committed by GitHub
parent cf5db95953
commit daa4600fd3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/google_utils.py
View File

@ -27,7 +27,7 @@ def attempt_download(file, repo='ultralytics/yolov3'):
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('utf-8').split('\n')[-2] 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: