Update install.bat

This commit is contained in:
Jeffreytsai1004 2025-02-16 17:17:19 +08:00
parent b223d55cc2
commit 0b34fec80b

View File

@ -185,6 +185,55 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
exit /b 1
)
REM 检查模型文件是否存在
IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\diffusion_prior\dif_ckpt.pt" (
echo ERROR: Neural Haircut Diffusion Prior model not found
echo Please run download_resource.bat first
exit /b 1
)
IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\strand_prior\strand_ckpt.pt" (
echo ERROR: Neural Haircut Strand Prior model not found
echo Please run download_resource.bat first
exit /b 1
)
IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\pixie_data\pixie_data" (
echo ERROR: PIXIE model not found
echo Please run download_resource.bat first
exit /b 1
)
IF NOT EXIST "%PROJECT_DIR%\resource\Matte-Anything\sam_vit_h_4b8939.pth" (
echo ERROR: SAM model not found
echo Please run download_resource.bat first
exit /b 1
)
IF NOT EXIST "%PROJECT_DIR%\resource\Matte-Anything\groundingdino_swint_ogc.pth" (
echo ERROR: GroundingDINO model not found
echo Please run download_resource.bat first
exit /b 1
)
IF NOT EXIST "%PROJECT_DIR%\resource\Matte-Anything\model.pth" (
echo ERROR: Matte-Anything model not found
echo Please run download_resource.bat first
exit /b 1
)
IF NOT EXIST "%PROJECT_DIR%\resource\openpose\models\pose\coco\pose_iter_584000.caffemodel" (
echo ERROR: OpenPose model not found
echo Please run download_resource.bat first
exit /b 1
)
IF NOT EXIST "%PROJECT_DIR%\resource\hyperIQA\pretrained\hyperIQA.pth" (
echo ERROR: hyperIQA model not found
echo Please run download_resource.bat first
exit /b 1
)
REM 复制模型文件到对应位置
echo 正在复制模型文件...