From 30dbf101b1917aad6c32f3217b03f13213a9e06c Mon Sep 17 00:00:00 2001
From: Jeffreytsai1004 <jeffreytsai1004@gmail.com>
Date: Sun, 16 Feb 2025 15:43:26 +0800
Subject: [PATCH] Update

---
 download_resource.bat | 8 ++++----
 install.bat           | 9 +++++----
 2 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/download_resource.bat b/download_resource.bat
index 706bbf1..a499162 100644
--- a/download_resource.bat
+++ b/download_resource.bat
@@ -15,11 +15,11 @@ IF EXIST "hyperIQA" rmdir /s /q "hyperIQA"
 
 REM 定义模型信息
 SET "MODEL_NEURAL_HAIRCUT_NAME=Neural Haircut模型"
-SET "MODEL_NEURAL_HAIRCUT_PATH=resource\NeuralHaircut\diffusion_prior\model.pt"
+SET "MODEL_NEURAL_HAIRCUT_PATH=resource\NeuralHaircut\pretrained_models\diffusion_prior\model.pt"
 SET "MODEL_NEURAL_HAIRCUT_URL=1_9EOUXHayKiGH5nkrayncln3d6m1uV7f"
 
 SET "MODEL_PIXIE_NAME=PIXIE模型数据"
-SET "MODEL_PIXIE_PATH=resource\NeuralHaircut\PIXIE\pixie_data"
+SET "MODEL_PIXIE_PATH=resource\NeuralHaircut\PIXIE\pixie_data.tar.gz"
 SET "MODEL_PIXIE_URL=1mPcGu62YPc4MdkT8FFiOCP629xsENHZf"
 
 SET "MODEL_PIXIE_FACE_NAME=PIXIE Face模型"
@@ -112,7 +112,7 @@ echo.
 REM 创建resource目录结构
 mkdir resource 2>nul
 cd resource
-mkdir NeuralHaircut\diffusion_prior 2>nul
+mkdir NeuralHaircut\pretrained_models\diffusion_prior 2>nul
 mkdir NeuralHaircut\PIXIE 2>nul
 mkdir Matte-Anything 2>nul
 mkdir openpose\models\pose\coco 2>nul
@@ -213,7 +213,7 @@ IF EXIST "%MODEL_NEURAL_HAIRCUT_PATH%" (
     echo 跳过%MODEL_NEURAL_HAIRCUT_NAME%下载...
 ) ELSE (
     echo 正在下载%MODEL_NEURAL_HAIRCUT_NAME%...
-    cd diffusion_prior
+    cd pretrained_models\diffusion_prior
     python -m gdown "%MODEL_NEURAL_HAIRCUT_URL%"
     cd ..\..
 )
diff --git a/install.bat b/install.bat
index 110d55d..f18d788 100644
--- a/install.bat
+++ b/install.bat
@@ -152,8 +152,9 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
     echo.
     echo resource/
     echo ├── NeuralHaircut/
-    echo │   ├── diffusion_prior/
-    echo │   │   └── model.pt
+    echo │   ├── pretrained_models/
+    echo │   │   └── diffusion_prior/
+    echo │   │       └── model.pt
     echo │   └── PIXIE/
     echo │       └── pixie_data/
     echo ├── Matte-Anything/
@@ -175,14 +176,14 @@ REM 复制模型文件到对应位置
 echo 正在复制模型文件...
 
 REM 创建必要的目录
-mkdir "%PROJECT_DIR%\ext\NeuralHaircut\diffusion_prior" 2>nul
+mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior" 2>nul
 mkdir "%PROJECT_DIR%\ext\PIXIE" 2>nul
 mkdir "%PROJECT_DIR%\ext\Matte-Anything\pretrained" 2>nul
 mkdir "%PROJECT_DIR%\ext\openpose\models\pose\coco" 2>nul
 mkdir "%PROJECT_DIR%\ext\hyperIQA\pretrained" 2>nul
 
 REM Neural Haircut
-xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\diffusion_prior\model.pt" "%PROJECT_DIR%\ext\NeuralHaircut\diffusion_prior\"
+xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior\model.pt" "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior\"
 
 REM PIXIE
 cd "%PROJECT_DIR%\ext\PIXIE"