fixed the problem of downloading models ()

pull/57/merge
spacewalk01 2022-08-08 16:17:40 +09:00 committed by GitHub
parent 8758cdd838
commit c14ba0c297
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

View File

@ -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