diff --git a/download_resource.bat b/download_resource.bat index 49a3447..f79c258 100644 --- a/download_resource.bat +++ b/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" -cd diffusion_prior -python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f" -cd ..\PIXIE -python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf" - -REM 使用7-Zip解压tar.gz文件 -IF NOT EXIST pixie_data.tar.gz ( - echo 错误:下载pixie_data.tar.gz失败 - exit /b 1 +IF EXIST "diffusion_prior\model.pt" ( + echo Neural Haircut模型已存在,跳过下载... +) ELSE ( + python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8" +) + +cd diffusion_prior +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 -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 -python -m gdown "1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW" -O model.pth --retry-times 5 +IF EXIST "sam_vit_h_4b8939.pth" ( + echo SAM模型已存在,跳过下载... +) 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 NOT EXIST models.tar.gz ( - echo ERROR: Failed to download models.tar.gz - exit /b 1 +IF EXIST "models\pose_iter_584000.caffemodel" ( + echo OpenPose模型已存在,跳过下载... +) 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.