This commit is contained in:
Jeffreytsai1004 2025-02-16 15:23:10 +08:00
parent d8006172d3
commit f31b034c60
2 changed files with 11 additions and 11 deletions

View File

@ -19,9 +19,13 @@ SET "MODEL_NEURAL_HAIRCUT_PATH=resource\NeuralHaircut\pretrained_models\diffusio
SET "MODEL_NEURAL_HAIRCUT_URL=1_9EOUXHayKiGH5nkrayncln3d6m1uV7f"
SET "MODEL_PIXIE_NAME=PIXIE模型数据"
SET "MODEL_PIXIE_PATH=resource\PIXIE\pixie_data.tar.gz"
SET "MODEL_PIXIE_PATH=resource\PIXIE\pixie_data"
SET "MODEL_PIXIE_URL=1mPcGu62YPc4MdkT8FFiOCP629xsENHZf"
SET "MODEL_PIXIE_FACE_NAME=PIXIE Face模型"
SET "MODEL_PIXIE_FACE_PATH=resource\PIXIE\pixie_data\pixie_data"
SET "MODEL_PIXIE_FACE_URL=https://github.com/yfeng95/PIXIE/raw/master/fetch_model.sh"
SET "MODEL_MATTE_NAME=Matte-Anything模型"
SET "MODEL_MATTE_PATH=resource\Matte-Anything\model.pth"
SET "MODEL_MATTE_URL=1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW"
@ -214,22 +218,16 @@ IF EXIST "%MODEL_NEURAL_HAIRCUT_PATH%" (
cd ..\..
)
REM 下载 PIXIE 文件
cd PIXIE
IF EXIST "pixie_data.tar" (
echo PIXIE模型已存在跳过下载...
) ELSE (
echo 正在下载PIXIE模型...
python -m gdown "%MODEL_PIXIE_URL%"
REM 使用7-Zip解压tar.gz文件
IF NOT EXIST pixie_data.tar.gz (
echo 错误下载pixie_data.tar.gz失败
exit /b 1
)
7z x pixie_data.tar.gz -y
7z x pixie_data.tar -y
del pixie_data.tar.gz
del pixie_data.tar
echo 正在下载PIXIE Face模型...
curl -L "%MODEL_PIXIE_FACE_URL%" -o fetch_model.sh
bash fetch_model.sh
)
REM 下载 Matte-Anything 文件

View File

@ -135,6 +135,8 @@ CALL "%MICROMAMBA_EXE%" create -p %MAMBA_ROOT_PREFIX%\envs\pixie-env python=3.8
REM 安装 gaussian_splatting_hair 环境
CALL "%MICROMAMBA_EXE%" activate -p %MAMBA_ROOT_PREFIX%\envs\gaussian_splatting_hair
pip install -r requirements.txt
REM 安装外部库
cd %PROJECT_DIR%\ext\pytorch3d
pip install -e .
cd %PROJECT_DIR%\ext\NeuralHaircut\npbgpp