Update download_resource.bat
This commit is contained in:
parent
aecd119dfd
commit
103ca1e623
@ -170,104 +170,96 @@ echo * 请耐心等待...
|
||||
echo **************************************************************************************************
|
||||
echo.
|
||||
|
||||
echo [1/8] 正在下载Neural Haircut文件...
|
||||
mkdir %PROJECT_DIR%/resource/NeuralHaircut
|
||||
REM 根据之前的验证结果下载缺失的模型文件
|
||||
IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\dif_ckpt.pt" (
|
||||
echo [1/!MISSING_FILES!] 正在下载Neural Haircut Diffusion Prior模型...
|
||||
mkdir %PROJECT_DIR%/resource/NeuralHaircut/pretrained_models/diffusion_prior
|
||||
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 检查下载后的文件完整性
|
||||
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 [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
|
||||
)
|
||||
)
|
||||
|
||||
echo [2/8] 正在下载Neural HaircutPIXIE模型文件...
|
||||
IF EXIST "%PIXIE_PATH%" (
|
||||
echo PIXIE模型已存在，跳过下载...
|
||||
) ELSE (
|
||||
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扩散先验模型文件...
|
||||
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
|
||||
IF EXIST "sam_vit_h_4b8939.pth" (
|
||||
echo SAM模型已存在，跳过下载...
|
||||
) ELSE (
|
||||
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%
|
||||
)
|
||||
|
||||
echo [7/8] 正在下载OpenPose模型文件...
|
||||
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
|
||||
IF EXIST "pose_iter_584000.caffemodel" (
|
||||
echo OpenPose模型已存在，跳过下载...
|
||||
) ELSE (
|
||||
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%
|
||||
)
|
||||
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user