mirror of https://github.com/WongKinYiu/yolov7.git
fixed the problem of downloading models (#451)
parent
8758cdd838
commit
c14ba0c297
utils
|
@ -26,7 +26,8 @@ def attempt_download(file, repo='WongKinYiu/yolov7'):
|
|||
assets = [x['name'] for x in response['assets']] # release assets
|
||||
tag = response['tag_name'] # i.e. 'v1.0'
|
||||
except: # fallback plan
|
||||
assets = ['yolov7.pt']
|
||||
assets = ['yolov7.pt', 'yolov7-tiny.pt', 'yolov7x.pt', 'yolov7-d6.pt', 'yolov7-e6.pt',
|
||||
'yolov7-e6e.pt', 'yolov7-w6.pt']
|
||||
tag = subprocess.check_output('git tag', shell=True).decode().split()[-1]
|
||||
|
||||
name = file.name
|
||||
|
|
Loading…
Reference in New Issue