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:36:29 +08:00
parent f0e6d9c90d
commit 2166c44293
2 changed files with 13 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
download_resource.bat
View File

@ -15,15 +15,15 @@ IF EXIST "hyperIQA" rmdir /s /q "hyperIQA"
REM 定义模型信息
SET "MODEL_NEURAL_HAIRCUT_NAME=Neural Haircut模型"
SET "MODEL_NEURAL_HAIRCUT_PATH=resource\NeuralHaircut\pretrained_models\diffusion_prior\model.pt"
SET "MODEL_NEURAL_HAIRCUT_PATH=resource\NeuralHaircut\diffusion_prior\model.pt"
SET "MODEL_NEURAL_HAIRCUT_URL=1_9EOUXHayKiGH5nkrayncln3d6m1uV7f"
SET "MODEL_PIXIE_NAME=PIXIE模型数据"
SET "MODEL_PIXIE_PATH=resource\PIXIE\pixie_data"
SET "MODEL_PIXIE_PATH=resource\NeuralHaircut\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_PATH=resource\NeuralHaircut\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模型"
@ -112,8 +112,8 @@ echo.
REM 创建resource目录结构
mkdir resource 2>nul
cd resource
mkdir NeuralHaircut\pretrained_models\diffusion_prior 2>nul
mkdir PIXIE 2>nul
mkdir NeuralHaircut\diffusion_prior 2>nul
mkdir NeuralHaircut\PIXIE 2>nul
mkdir Matte-Anything 2>nul
mkdir openpose\models\pose\coco 2>nul
mkdir hyperIQA\pretrained 2>nul
@ -219,7 +219,7 @@ IF EXIST "%MODEL_NEURAL_HAIRCUT_PATH%" (
)
REM 下载 PIXIE 文件
cd PIXIE
cd NeuralHaircut\PIXIE
IF EXIST "pixie_data.tar" (
echo PIXIE模型已存在，跳过下载...
) ELSE (

15
install.bat
View File

@ -152,11 +152,10 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
echo.
echo resource/
echo ├── NeuralHaircut/
echo │ ├── pretrained_models/
echo │ │ └── diffusion_prior/
echo │ │ └── model.pt
echo ├── PIXIE/
echo │ └── pixie_data/
echo │ ├── diffusion_prior/
echo │ │ └── model.pt
echo │ └── PIXIE/
echo │ └── pixie_data/
echo ├── Matte-Anything/
echo │ ├── sam_vit_h_4b8939.pth
echo │ ├── groundingdino_swint_ogc.pth
@ -176,18 +175,18 @@ REM 复制模型文件到对应位置
echo 正在复制模型文件...
REM 创建必要的目录
mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior" 2>nul
mkdir "%PROJECT_DIR%\ext\NeuralHaircut\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\pretrained_models\diffusion_prior\model.pt" "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior\"
xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\diffusion_prior\model.pt" "%PROJECT_DIR%\ext\NeuralHaircut\diffusion_prior\"
REM PIXIE
cd "%PROJECT_DIR%\ext\PIXIE"
7z x "%PROJECT_DIR%\resource\PIXIE\pixie_data.tar.gz" -y
7z x "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\pixie_data.tar.gz" -y
7z x pixie_data.tar -y
del pixie_data.tar