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-17 23:29:42 +08:00
parent f7e08c8047
commit ce19540e82
2 changed files with 33 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
download_resource.bat
View File

@ -15,12 +15,6 @@ python -m pip install --upgrade pip
REM 安装gdown
python -m pip install gdown
REM 安装7z
python -m pip install py7zr
REM 安装tar
python -m pip install tar
mkdir %PROJECT_DIR%\resource\NeuralHaircut\
cd %PROJECT_DIR%\resource\NeuralHaircut\
python -m gdown --folder https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8

46
install.bat
View File

@ -186,27 +186,47 @@ REM 复制模型文件到对应位置
echo 正在复制模型文件...
REM 创建必要的目录
mkdir "%PROJECT_DIR%\ext\NeuralHaircut" 2>nul
mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models" 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\NeuralHaircut\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\openpose\models" 2>nul
mkdir "%PROJECT_DIR%\ext\hyperIQA\pretrained" 2>nul
REM 复制NeuralHaircut相关文件
xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\*" "%PROJECT_DIR%\ext\NeuralHaircut\"
xcopy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\*" "%PROJECT_DIR%\ext\PIXIE"
REM 复制 NeuralHaircut 相关文件
xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior\dif_ckpt.pt" ^
"%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior\"
xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\strand_prior\strand_ckpt.pt" ^
"%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\strand_prior\"
xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\pixie_data.tar.gz" ^
"%PROJECT_DIR%\ext\NeuralHaircut\PIXIE\"
REM 复制Matte-Anything相关文件
xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\*" "%PROJECT_DIR%\ext\Matte-Anything\"
REM 复制 Matte-Anything 相关文件
xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\pretrained\ViTMatte_B_DIS.pth" ^
"%PROJECT_DIR%\ext\Matte-Anything\pretrained\"
REM 复制OpenPose相关文件
xcopy /E /I /Y "%PROJECT_DIR%\resource\openpose\*" "%PROJECT_DIR%\ext\openpose\"
REM 复制 OpenPose 相关文件
xcopy /Y "%PROJECT_DIR%\resource\openpose\models\models.tar.gz" ^
"%PROJECT_DIR%\ext\openpose\models\"
REM 复制hyperIQA相关文件
xcopy /Y "%PROJECT_DIR%\resource\hyperIQA\*" "%PROJECT_DIR%\ext\hyperIQA\"
REM 复制 hyperIQA 相关文件
xcopy /Y "%PROJECT_DIR%\resource\hyperIQA\pretrained\koniq_pretrained.pkl" ^
"%PROJECT_DIR%\ext\hyperIQA\pretrained\"
REM 解压缩文件
cd "%PROJECT_DIR%\ext\NeuralHaircut\PIXIE"
if exist "pixie_data.tar.gz" (
tar -xvzf pixie_data.tar.gz
del pixie_data.tar.gz
)
cd "%PROJECT_DIR%\ext\openpose\models"
if exist "models.tar.gz" (
tar -xvzf models.tar.gz
del models.tar.gz
)
cd "%PROJECT_DIR%"
REM 安装 PIXIE 环境
CALL activate_pixie-env.bat