CGNICO/GaussianHaircut
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update

Browse Source
This commit is contained in:
Jeffreytsai1004 2025-02-16 15:43:26 +08:00
parent 2166c44293
commit 30dbf101b1
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
download_resource.bat
View File

@ -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 ..\..
)

9
install.bat
View File

@ -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"