Update download_resource.bat

download_resource.bat

@@ -70,46 +70,80 @@ echo ************************************************************
 REM 下载 Neural Haircut 文件
 cd NeuralHaircut
 echo 正在下载Neural Haircut文件...
-python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8"
+IF EXIST "diffusion_prior\model.pt" (
-cd diffusion_prior
+    echo Neural Haircut模型已存在，跳过下载...
-python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f"
+) ELSE (
-cd ..\PIXIE
+    python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8"
-python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf"
+)
 
-REM 使用7-Zip解压tar.gz文件
+cd diffusion_prior
-IF NOT EXIST pixie_data.tar.gz (
+IF EXIST "model.pt" (
+    echo 扩散先验模型已存在，跳过下载...
+) ELSE (
+    python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f"
+)
+
+cd ..\PIXIE
+IF EXIST "pixie_data.tar" (
+    echo PIXIE模型已存在，跳过下载...
+) ELSE (
+    python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf"
+    
+    REM 使用7-Zip解压tar.gz文件
+    IF NOT EXIST pixie_data.tar.gz (
         echo 错误：下载pixie_data.tar.gz失败
         exit /b 1
+    )
+    7z x pixie_data.tar.gz -y
+    7z x pixie_data.tar -y
+    del pixie_data.tar.gz
+    del pixie_data.tar
 )
-7z x pixie_data.tar.gz -y
-7z x pixie_data.tar -y
-del pixie_data.tar.gz
-del pixie_data.tar
 
 REM 下载 Matte-Anything 文件
 cd ..\..\Matte-Anything
 echo 正在下载Matte-Anything文件...
-curl -L --retry 5 -o sam_vit_h_4b8939.pth https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth
+IF EXIST "sam_vit_h_4b8939.pth" (
-curl -L --retry 5 -o groundingdino_swint_ogc.pth https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth
+    echo SAM模型已存在，跳过下载...
-python -m gdown "1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW" -O model.pth --retry-times 5
+) ELSE (
+    curl -L --retry 5 -o sam_vit_h_4b8939.pth https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth
+)
+IF EXIST "groundingdino_swint_ogc.pth" (
+    echo GroundingDINO模型已存在，跳过下载...
+) ELSE (
+    curl -L --retry 5 -o groundingdino_swint_ogc.pth https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth
+)
+IF EXIST "model.pth" (
+    echo Matte-Anything模型已存在，跳过下载...
+) ELSE (
+    python -m gdown "1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW" -O model.pth
+)
 
 REM 下载 OpenPose 模型
 cd ..\openpose
 echo 正在下载OpenPose模型...
-python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" -O models.tar.gz
+IF EXIST "models\pose_iter_584000.caffemodel" (
-IF NOT EXIST models.tar.gz (
+    echo OpenPose模型已存在，跳过下载...
-    echo ERROR: Failed to download models.tar.gz
+) ELSE (
+    python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" -O models.tar.gz
+    IF NOT EXIST models.tar.gz (
+        echo 错误：下载models.tar.gz失败
         exit /b 1
+    )
+    7z x models.tar.gz -y
+    7z x models.tar -y -omodels
+    del models.tar.gz
+    del models.tar
 )
-7z x models.tar.gz -y
-7z x models.tar -y -omodels
-del models.tar.gz
-del models.tar
 
 REM 下载 hyperIQA 模型
 cd ..\hyperIQA\pretrained
 echo 正在下载hyperIQA模型...
-python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE"
+IF EXIST "hyperIQA.pth" (
+    echo hyperIQA模型已存在，跳过下载...
+) ELSE (
+    python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE"
+)
 
 cd ..\..
 echo.