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 15:43:26 +08:00
parent 2166c44293
commit 30dbf101b1
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
download_resource.bat
View File

@ -15,11 +15,11 @@ IF EXIST "hyperIQA" rmdir /s /q "hyperIQA"
REM 定义模型信息 REM 定义模型信息
SET "MODEL_NEURAL_HAIRCUT_NAME=Neural Haircut模型" 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_NEURAL_HAIRCUT_URL=1_9EOUXHayKiGH5nkrayncln3d6m1uV7f"
SET "MODEL_PIXIE_NAME=PIXIE模型数据" SET "MODEL_PIXIE_NAME=PIXIE模型数据"
SET "MODEL_PIXIE_PATH=resource\NeuralHaircut\PIXIE\pixie_data" SET "MODEL_PIXIE_PATH=resource\NeuralHaircut\PIXIE\pixie_data.tar.gz"
SET "MODEL_PIXIE_URL=1mPcGu62YPc4MdkT8FFiOCP629xsENHZf" SET "MODEL_PIXIE_URL=1mPcGu62YPc4MdkT8FFiOCP629xsENHZf"
SET "MODEL_PIXIE_FACE_NAME=PIXIE Face模型" SET "MODEL_PIXIE_FACE_NAME=PIXIE Face模型"
@ -112,7 +112,7 @@ echo.
REM 创建resource目录结构 REM 创建resource目录结构
mkdir resource 2>nul mkdir resource 2>nul
cd resource cd resource
mkdir NeuralHaircut\diffusion_prior 2>nul mkdir NeuralHaircut\pretrained_models\diffusion_prior 2>nul
mkdir NeuralHaircut\PIXIE 2>nul mkdir NeuralHaircut\PIXIE 2>nul
mkdir Matte-Anything 2>nul mkdir Matte-Anything 2>nul
mkdir openpose\models\pose\coco 2>nul mkdir openpose\models\pose\coco 2>nul
@ -213,7 +213,7 @@ IF EXIST "%MODEL_NEURAL_HAIRCUT_PATH%" (
echo 跳过%MODEL_NEURAL_HAIRCUT_NAME%下载... echo 跳过%MODEL_NEURAL_HAIRCUT_NAME%下载...
) ELSE ( ) ELSE (
echo 正在下载%MODEL_NEURAL_HAIRCUT_NAME%... echo 正在下载%MODEL_NEURAL_HAIRCUT_NAME%...
cd diffusion_prior cd pretrained_models\diffusion_prior
python -m gdown "%MODEL_NEURAL_HAIRCUT_URL%" python -m gdown "%MODEL_NEURAL_HAIRCUT_URL%"
cd ..\.. cd ..\..
) )

7
install.bat
View File

@ -152,7 +152,8 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
echo. echo.
echo resource/ echo resource/
echo ├── NeuralHaircut/ echo ├── NeuralHaircut/
echo │ ├── diffusion_prior/ echo │ ├── pretrained_models/
echo │ │ └── diffusion_prior/
echo │ │ └── model.pt echo │ │ └── model.pt
echo │ └── PIXIE/ echo │ └── PIXIE/
echo │ └── pixie_data/ echo │ └── pixie_data/
@ -175,14 +176,14 @@ REM 复制模型文件到对应位置
echo 正在复制模型文件... echo 正在复制模型文件...
REM 创建必要的目录 REM 创建必要的目录
mkdir "%PROJECT_DIR%\ext\NeuralHaircut\diffusion_prior" 2>nul mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_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
mkdir "%PROJECT_DIR%\ext\hyperIQA\pretrained" 2>nul mkdir "%PROJECT_DIR%\ext\hyperIQA\pretrained" 2>nul
REM Neural Haircut REM Neural Haircut
xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\diffusion_prior\model.pt" "%PROJECT_DIR%\ext\NeuralHaircut\diffusion_prior\" xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior\model.pt" "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior\"
REM PIXIE REM PIXIE
cd "%PROJECT_DIR%\ext\PIXIE" cd "%PROJECT_DIR%\ext\PIXIE"