diff --git a/download_resource.bat b/download_resource.bat index 6c87751..93c5812 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -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 文件 diff --git a/install.bat b/install.bat index 47ccfbf..c7ea525 100644 --- a/install.bat +++ b/install.bat @@ -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