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 17:28:53 +08:00
parent 0b34fec80b
commit 2d7453bbad
2 changed files with 18 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
download_resource.bat
View File

@ -98,6 +98,7 @@ REM 创建resource目录结构
mkdir resource 2>nul
cd resource
mkdir NeuralHaircut 2>nul
mkdir NeuralHaircut\pretrained_models 2>nul
mkdir Matte-Anything 2>nul
mkdir openpose 2>nul
mkdir hyperIQA 2>nul
@ -206,8 +207,14 @@ cd resource\NeuralHaircut
echo 正在下载Neural Haircut预训练文件夹...
python -m gdown --folder "%NEURAL_HAIRCUT_FOLDER_URL%"
REM 更新 diffusion_prior 模型
cd pretrained_models\diffusion_prior
echo 正在更新Neural Haircut Diffusion Prior模型...
python -m gdown "%NEURAL_HAIRCUT_MODEL_URL%"
cd ..\..
REM 下载 PIXIE 文件
cd NeuralHaircut\PIXIE
cd PIXIE
IF EXIST "pixie_data.tar" (
echo PIXIE模型已存在，跳过下载...
) ELSE (

18
install.bat
View File

@ -164,10 +164,11 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
echo.
echo resource/
echo ├── NeuralHaircut/
echo │ ├── diffusion_prior/
echo │ │ └── dif_ckpt.pt
echo │ ├── strand_prior/
echo │ │ └── strand_ckpt.pt
echo │ ├── pretrained_models/
echo │ │ ├── diffusion_prior/
echo │ │ │ └── dif_ckpt.pt
echo │ │ └── strand_prior/
echo │ │ └── strand_ckpt.pt
echo │ └── PIXIE/
echo │ └── pixie_data/
echo ├── Matte-Anything/
@ -186,13 +187,13 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
)
REM 检查模型文件是否存在
IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\diffusion_prior\dif_ckpt.pt" (
IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\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" (
IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\strand_prior\strand_ckpt.pt" (
echo ERROR: Neural Haircut Strand Prior model not found
echo Please run download_resource.bat first
exit /b 1
@ -239,8 +240,9 @@ echo 正在复制模型文件...
REM 创建必要的目录
mkdir "%PROJECT_DIR%\ext\NeuralHaircut" 2>nul
mkdir "%PROJECT_DIR%\ext\NeuralHaircut\diffusion_prior" 2>nul
mkdir "%PROJECT_DIR%\ext\NeuralHaircut\strand_prior" 2>nul
mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models" 2>nul
mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior" 2>nul
mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\strand_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