From 0b34fec80bb33e90b0da9b2b132989a17eb90313 Mon Sep 17 00:00:00 2001
From: Jeffreytsai1004 <jeffreytsai1004@gmail.com>
Date: Sun, 16 Feb 2025 17:17:19 +0800
Subject: [PATCH] Update install.bat

---
 install.bat | 49 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)

diff --git a/install.bat b/install.bat
index 814e1b6..90dc3cd 100644
--- a/install.bat
+++ b/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 正在复制模型文件...