diff --git a/download_resource.bat b/download_resource.bat index 82c027f..fbe2952 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -170,104 +170,96 @@ echo * 请耐心等待... echo ************************************************************************************************** echo. -echo [1/8] 正在下载Neural Haircut文件... -mkdir %PROJECT_DIR%/resource/NeuralHaircut -cd %PROJECT_DIR%/resource/NeuralHaircut -IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\dif_ckpt.pt" ( - IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\strand_ckpt.pt" ( - echo Neural Haircut文件已存在,跳过下载... - ) ELSE ( - echo Neural Haircut Strand Prior模型不完整,重新下载... - python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8" - ) -) ELSE ( - echo Neural Haircut Diffusion Prior模型不完整,重新下载... - python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8" -) - -REM 检查下载后的文件完整性 +REM 根据之前的验证结果下载缺失的模型文件 IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\dif_ckpt.pt" ( - echo 错误:Neural Haircut Diffusion Prior模型下载失败 - exit /b 1 -) -IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\strand_ckpt.pt" ( - echo 错误:Neural Haircut Strand Prior模型下载失败 - exit /b 1 + echo [1/!MISSING_FILES!] 正在下载Neural Haircut Diffusion Prior模型... + mkdir %PROJECT_DIR%/resource/NeuralHaircut/pretrained_models/diffusion_prior + cd %PROJECT_DIR%/resource/NeuralHaircut + python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8" + IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\dif_ckpt.pt" ( + echo 错误:Neural Haircut Diffusion Prior模型下载失败 + exit /b 1 + ) ) -echo [2/8] 正在下载Neural HaircutPIXIE模型文件... -IF EXIST "%PIXIE_PATH%" ( - echo PIXIE模型已存在,跳过下载... -) ELSE ( +IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\strand_ckpt.pt" ( + echo [2/!MISSING_FILES!] 正在下载Neural Haircut Strand Prior模型... + mkdir %PROJECT_DIR%/resource/NeuralHaircut/pretrained_models/strand_prior + cd %PROJECT_DIR%/resource/NeuralHaircut + python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8" + IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\strand_ckpt.pt" ( + echo 错误:Neural Haircut Strand Prior模型下载失败 + exit /b 1 + ) +) + +IF NOT EXIST "%PIXIE_PATH%" ( + echo [3/!MISSING_FILES!] 正在下载PIXIE模型... mkdir %PROJECT_DIR%/resource/NeuralHaircut/PIXIE cd %PROJECT_DIR%/resource/NeuralHaircut/PIXIE python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf" && tar -xvzf pixie_data.tar.gz ./ && rm pixie_data.tar.gz cd %PROJECT_DIR% -) - -echo [3/8] 正在下载hyperIQA模型文件... -mkdir %PROJECT_DIR%/resource/hyperIQA/pretrained -cd %PROJECT_DIR%/resource/hyperIQA/pretrained -IF EXIST "hyperIQA.pth" ( - echo hyperIQA模型已存在,跳过下载... -) ELSE ( - python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE" - IF NOT EXIST "hyperIQA.pth" ( - echo 错误:hyperIQA模型下载失败 + IF NOT EXIST "%PIXIE_PATH%" ( + echo 错误:PIXIE模型下载失败 exit /b 1 ) ) -cd %PROJECT_DIR% -echo [4/8] 正在下载Matte-Anything扩散先验模型文件... -mkdir %PROJECT_DIR%/resource/Matte-Anything/pretrained -cd %PROJECT_DIR%/resource/Matte-Anything/pretrained -IF EXIST "sam_vit_h_4b8939.pth" ( - echo SAM模型已存在,跳过下载... -) ELSE ( +IF NOT EXIST "%HYPERIQA_PATH%" ( + echo [4/!MISSING_FILES!] 正在下载hyperIQA模型... + mkdir %PROJECT_DIR%/resource/hyperIQA/pretrained + cd %PROJECT_DIR%/resource/hyperIQA/pretrained + python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE" + IF NOT EXIST "%HYPERIQA_PATH%" ( + echo 错误:hyperIQA模型下载失败 + exit /b 1 + ) + cd %PROJECT_DIR% +) + +IF NOT EXIST "%SAM_PATH%" ( + echo [5/!MISSING_FILES!] 正在下载SAM模型... + mkdir %PROJECT_DIR%/resource/Matte-Anything/pretrained + cd %PROJECT_DIR%/resource/Matte-Anything/pretrained wget "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth" - IF NOT EXIST "sam_vit_h_4b8939.pth" ( + IF NOT EXIST "%SAM_PATH%" ( echo 错误:SAM模型下载失败 exit /b 1 ) ) -echo [5/8] 正在下载Matte-AnythingGroundingDINO模型文件... -IF EXIST "groundingdino_swint_ogc.pth" ( - echo GroundingDINO模型已存在,跳过下载... -) ELSE ( +IF NOT EXIST "%DINO_PATH%" ( + echo [6/!MISSING_FILES!] 正在下载GroundingDINO模型... + cd %PROJECT_DIR%/resource/Matte-Anything/pretrained wget "https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth" - IF NOT EXIST "groundingdino_swint_ogc.pth" ( + IF NOT EXIST "%DINO_PATH%" ( echo 错误:GroundingDINO模型下载失败 exit /b 1 ) ) -echo [6/8] 正在下载Matte-Anything模型文件... -IF EXIST "model.pth" ( - echo Matte-Anything模型已存在,跳过下载... -) ELSE ( +IF NOT EXIST "%MATTE_PATH%" ( + echo [7/!MISSING_FILES!] 正在下载Matte-Anything模型... + cd %PROJECT_DIR%/resource/Matte-Anything/pretrained python -m gdown "1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW" - IF NOT EXIST "model.pth" ( + IF NOT EXIST "%MATTE_PATH%" ( echo 错误:Matte-Anything模型下载失败 exit /b 1 ) + cd %PROJECT_DIR% ) -cd %PROJECT_DIR% -echo [7/8] 正在下载OpenPose模型文件... -mkdir %PROJECT_DIR%/resource/openpose/models/pose/coco -cd %PROJECT_DIR%/resource/openpose/models/pose/coco -IF EXIST "pose_iter_584000.caffemodel" ( - echo OpenPose模型已存在,跳过下载... -) ELSE ( +IF NOT EXIST "%OPENPOSE_PATH%" ( + echo [8/!MISSING_FILES!] 正在下载OpenPose模型... + mkdir %PROJECT_DIR%/resource/openpose/models/pose/coco + cd %PROJECT_DIR%/resource/openpose/models/pose/coco python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" && tar -xvzf models.tar.gz && rm models.tar.gz - IF NOT EXIST "pose_iter_584000.caffemodel" ( + IF NOT EXIST "%OPENPOSE_PATH%" ( echo 错误:OpenPose模型下载失败 exit /b 1 ) + cd %PROJECT_DIR% ) -cd %PROJECT_DIR%