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

Update

Browse Source 
Update
This commit is contained in:
Jeffreytsai1004 2025-02-16 15:16:16 +08:00
parent 90318d973c
commit d8006172d3
2 changed files with 20 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
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\PIXIE\pixie_data.tar.gz"
SET "MODEL_PIXIE_URL=1mPcGu62YPc4MdkT8FFiOCP629xsENHZf"
SET "MODEL_MATTE_NAME=Matte-Anything模型"
@ -108,10 +108,10 @@ echo.
REM 创建resource目录结构
mkdir resource 2>nul
cd resource
mkdir NeuralHaircut\diffusion_prior 2>nul
mkdir NeuralHaircut\PIXIE 2>nul
mkdir NeuralHaircut\pretrained_models\diffusion_prior 2>nul
mkdir PIXIE 2>nul
mkdir Matte-Anything 2>nul
mkdir openpose\models\pose\coco 2>nul
mkdir openpose\models 2>nul
mkdir hyperIQA\pretrained 2>nul
REM 验证已有文件
@ -259,7 +259,7 @@ IF EXIST "%MODEL_MATTE_PATH%" (
)
REM 下载 OpenPose 模型
cd ..\openpose\models\pose\coco
cd ..\openpose\models
echo 正在下载OpenPose模型...
IF EXIST "%MODEL_OPENPOSE_PATH%" (
echo %MODEL_OPENPOSE_NAME%已存在，跳过下载...
@ -316,7 +316,7 @@ IF NOT EXIST "%MODEL_NEURAL_HAIRCUT_PATH%" (
IF NOT EXIST "%MODEL_PIXIE_PATH%" echo 警告：缺少%MODEL_PIXIE_NAME%
IF NOT EXIST "%MODEL_SAM_PATH%" echo 警告：缺少%MODEL_SAM_NAME%
IF NOT EXIST "%MODEL_DINO_PATH%" echo 警告：缺少%MODEL_DINO_NAME%
IF NOT EXIST "openpose\models\pose\coco" echo 警告：缺少OpenPose姿态模型
IF NOT EXIST "%MODEL_OPENPOSE_PATH%" echo 警告：缺少%MODEL_OPENPOSE_NAME%
IF NOT EXIST "%MODEL_HYPERIQA_PATH%" echo 警告：缺少%MODEL_HYPERIQA_NAME%
IF %VERIFY_FAILED% GTR 0 (

18
install.bat
View File

@ -167,11 +167,22 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
REM 复制模型文件到对应位置
echo 正在复制模型文件...
REM 创建必要的目录
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\pretrained_models\diffusion_prior\"
xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior\model.pt" "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior\"
REM PIXIE
xcopy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\pixie_data" "%PROJECT_DIR%\ext\PIXIE\"
cd "%PROJECT_DIR%\ext\PIXIE"
7z x "%PROJECT_DIR%\resource\PIXIE\pixie_data.tar.gz" -y
7z x pixie_data.tar -y
del pixie_data.tar
REM Matte-Anything
xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\sam_vit_h_4b8939.pth" "%PROJECT_DIR%\ext\Matte-Anything\pretrained\"
@ -191,9 +202,6 @@ echo 正在安装PIXIE依赖...
pip install pyyaml==5.4.1
pip install git+https://github.com/1adrianb/face-alignment.git@54623537fd9618ca7c15688fd85aba706ad92b59
REM 复制PIXIE数据
xcopy /E /I /Y "%PROJECT_DIR%\ext\PIXIE\*" "%PROJECT_DIR%\ext\NeuralHaircut\PIXIE\"
REM 安装 Matte-Anything 环境
CALL activate_matte_anything.bat
cd %PROJECT_DIR%\ext\Matte-Anything