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 08:44:22 +08:00
parent e91e16c4ca
commit f7e08c8047
3 changed files with 47 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
README_CN.md
View File

@ -47,20 +47,20 @@ resource/
├── NeuralHaircut/ ├── NeuralHaircut/
│ ├── pretrained_models/ │ ├── pretrained_models/
│ │ ├── diffusion_prior/ │ │ ├── diffusion_prior/
│ │ │ └── dif_ckpt.pt ## 扩散先验模型 │ │ │ └── dif_ckpt.pt # 扩散先验模型
│ │ └── strand_prior/ │ │ └── strand_prior/
│ │ └── strand_ckpt.pt ## 股线先验模型 │ │ └── strand_ckpt.pt # 股线先验模型
│ └── PIXIE/ │ └── PIXIE/
│ └── pixie_data.tar.gz │ └── pixie_data.tar.gz # PIXIE 模型数据存档
├── Matte-Anything/ ├── Matte-Anything/
│ └── pretrained/ │ └── pretrained/
│ └── ViTMatte_B_DIS.pth │ └── ViTMatte_B_DIS.pth # Matte-Anything 模型
├── openpose/ ├── openpose/
│ └── models/ │ └── models/
│ └── models.tar.gz │ └── models.tar.gz # OpenPose 模型
└── hyperIQA/ └── hyperIQA/
└── pretrained/ └── pretrained/
└── koniq_pretrained.pkl └── koniq_pretrained.pkl # 图像质量评估模型
``` ```
## 环境配置 ## 环境配置

23
download_resource.bat
View File

@ -2,6 +2,25 @@
set PROJECT_DIR=%~dp0 set PROJECT_DIR=%~dp0
cd %PROJECT_DIR% cd %PROJECT_DIR%
REM 检查是否安装了python
if not exist "%PROJECT_DIR%\venv\Scripts\python.exe" (
echo 错误：未找到python.exe
echo 请确保python已安装并配置好环境变量
exit /b 1
)
REM pip安装检查更新
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\ mkdir %PROJECT_DIR%\resource\NeuralHaircut\
cd %PROJECT_DIR%\resource\NeuralHaircut\ cd %PROJECT_DIR%\resource\NeuralHaircut\
python -m gdown --folder https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8 python -m gdown --folder https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8
@ -10,7 +29,7 @@ python -m gdown 1_9EOUXHayKiGH5nkrayncln3d6m1uV7f
mkdir %PROJECT_DIR%\resource\NeuralHaircut\PIXIE mkdir %PROJECT_DIR%\resource\NeuralHaircut\PIXIE
cd %PROJECT_DIR%\resource\NeuralHaircut\PIXIE cd %PROJECT_DIR%\resource\NeuralHaircut\PIXIE
python -m gdown 1mPcGu62YPc4MdkT8FFiOCP629xsENHZf python -m gdown 1mPcGu62YPc4MdkT8FFiOCP629xsENHZf
7z x pixie_data.tar.gz -o./ 7z x pixie_data.tar.gz -o./ && tar -xvzf pixie_data.tar.gz
mkdir %PROJECT_DIR%\resource\hyperIQA\pretrained mkdir %PROJECT_DIR%\resource\hyperIQA\pretrained
cd %PROJECT_DIR%\resource\hyperIQA\pretrained cd %PROJECT_DIR%\resource\hyperIQA\pretrained
python -m gdown 1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE python -m gdown 1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE
@ -24,4 +43,4 @@ python -m gdown 1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW
mkdir %PROJECT_DIR%\resource\openpose\ mkdir %PROJECT_DIR%\resource\openpose\
cd %PROJECT_DIR%\resource\openpose\ cd %PROJECT_DIR%\resource\openpose\
python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV && tar -xvzf models.tar.gz && rm models.tar.gz python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV && tar -xvzf models.tar.gz

33
install.bat
View File

@ -166,17 +166,22 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
echo ├── NeuralHaircut/ echo ├── NeuralHaircut/
echo │ ├── pretrained_models/ echo │ ├── pretrained_models/
echo │ │ ├── diffusion_prior/ echo │ │ ├── diffusion_prior/
echo │ │ │ └── dif_ckpt.pt echo │ │ │ └── dif_ckpt.pt # 扩散先验模型
echo │ │ └── strand_prior/ echo │ │ └── strand_prior/
echo │ │ └── strand_ckpt.pt echo │ │ └── strand_ckpt.pt # 股线先验模型
echo │ └── PIXIE/ echo │ └── PIXIE/
echo │ └── pixie_data/ echo │ └── pixie_data/ # PIXIE 模型数据存档
echo ├── Matte-Anything/ echo ├── Matte-Anything/
echo │ └── pretrained/
echo │ └── ViTMatte_B_DIS.pth # Matte-Anything 模型
echo ├── openpose/ echo ├── openpose/
echo │ └── models/
echo │ └── models/ # OpenPose 模型
echo └── hyperIQA/ echo └── hyperIQA/
echo └── pretrained/
echo └── koniq_pretrained.pkl # 图像质量评估模型
exit /b 1 exit /b 1
) )
REM 复制模型文件到对应位置 REM 复制模型文件到对应位置
echo 正在复制模型文件... echo 正在复制模型文件...
@ -190,11 +195,18 @@ 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
xcopy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\*" "%PROJECT_DIR%\ext\NeuralHaircut\" 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" xcopy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\*" "%PROJECT_DIR%\ext\PIXIE"
xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\*" "%PROJECT_DIR%\ext\Matte-Anything\pretrained\"
xcopy /Y "%PROJECT_DIR%\resource\openpose\*" "%PROJECT_DIR%\ext\openpose\*" REM 复制Matte-Anything相关文件
xcopy /Y "%PROJECT_DIR%\resource\hyperIQA\*" "%PROJECT_DIR%\ext\hyperIQA\*" xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\*" "%PROJECT_DIR%\ext\Matte-Anything\"
REM 复制OpenPose相关文件
xcopy /E /I /Y "%PROJECT_DIR%\resource\openpose\*" "%PROJECT_DIR%\ext\openpose\"
REM 复制hyperIQA相关文件
xcopy /Y "%PROJECT_DIR%\resource\hyperIQA\*" "%PROJECT_DIR%\ext\hyperIQA\"
REM 安装 PIXIE 环境 REM 安装 PIXIE 环境
CALL activate_pixie-env.bat CALL activate_pixie-env.bat
@ -215,11 +227,6 @@ pip install -e .
cd .. cd ..
pip install supervision==0.22.0 pip install supervision==0.22.0
mkdir pretrained 2>nul
cd pretrained
xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\*" .
cd ..
xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\model.pth" .
REM 安装 OpenPose REM 安装 OpenPose
CALL activate_openpose.bat CALL activate_openpose.bat