From 0d869161c22e45f2314969782f15b048b0886dcf Mon Sep 17 00:00:00 2001 From: darkliang <11710911@mail.sustech.edu.cn> Date: Wed, 8 Dec 2021 20:22:26 +0800 Subject: [PATCH] update download path --- misc/download_pretrained_model.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/misc/download_pretrained_model.py b/misc/download_pretrained_model.py index 2e48ead..c48bb6a 100644 --- a/misc/download_pretrained_model.py +++ b/misc/download_pretrained_model.py @@ -118,8 +118,11 @@ def download_file_and_uncompress(url, if __name__ == "__main__": - url = "https://github.com/ShiqiYu/OpenGait/releases/download/v1.0-beta/pretrained_model.zip" - download_file_and_uncompress( - url=url, extrapath='output') + urls = [ + "https://github.com/ShiqiYu/OpenGait/releases/download/v1.0/pretrained_casiab_model.zip", + "https://github.com/ShiqiYu/OpenGait/releases/download/v1.0/pretrained_oumvlp_model.zip"] + for url in urls: + download_file_and_uncompress( + url=url, extrapath='output') print("Pretrained model download success!")