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 mkdir resource 2>nul
cd resource cd resource
mkdir NeuralHaircut 2>nul mkdir NeuralHaircut 2>nul
mkdir NeuralHaircut\pretrained_models 2>nul
mkdir Matte-Anything 2>nul mkdir Matte-Anything 2>nul
mkdir openpose 2>nul mkdir openpose 2>nul
mkdir hyperIQA 2>nul mkdir hyperIQA 2>nul
@ -206,8 +207,14 @@ cd resource\NeuralHaircut
echo 正在下载Neural Haircut预训练文件夹... echo 正在下载Neural Haircut预训练文件夹...
python -m gdown --folder "%NEURAL_HAIRCUT_FOLDER_URL%" 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 文件 REM 下载 PIXIE 文件
cd NeuralHaircut\PIXIE cd PIXIE
IF EXIST "pixie_data.tar" ( IF EXIST "pixie_data.tar" (
echo PIXIE模型已存在，跳过下载... echo PIXIE模型已存在，跳过下载...
) ELSE ( ) ELSE (

18
install.bat
View File

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