From c14ba0c297b3b5fc0374c917db798c88f9dd226c Mon Sep 17 00:00:00 2001 From: spacewalk01 Date: Mon, 8 Aug 2022 16:17:40 +0900 Subject: [PATCH] fixed the problem of downloading models (#451) --- utils/google_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/utils/google_utils.py b/utils/google_utils.py index c311fd57..f363408e 100644 --- a/utils/google_utils.py +++ b/utils/google_utils.py @@ -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