update download path

ShiqiYu · Dec 8, 2021 · 0d86916 · 0d86916
1 parent bb6cd51
commit 0d86916
Showing 1 changed file with 6 additions and 3 deletions.
diff --git 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!")