Update google_utils.py

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

View File

@ -27,7 +27,7 @@ def attempt_download(file, repo='ultralytics/yolov3'):
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('utf-8').split('\n')[-2]
tag = subprocess.check_output('git tag', shell=True).decode().split()[-1]
name = file.name
if name in assets: