Update install.bat
This commit is contained in:
parent
b223d55cc2
commit
0b34fec80b
49
install.bat
49
install.bat
@ -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 正在复制模型文件...
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user