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 17:38:01 +08:00
parent 2d7453bbad
commit 4376af31d3
2 changed files with 13 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
download_resource.bat
View File

@ -24,7 +24,7 @@ SET "HYPERIQA_MODEL_URL=1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE"
REM 定义文件路径 REM 定义文件路径
SET "NEURAL_HAIRCUT_FOLDER_PATH=resource\NeuralHaircut" SET "NEURAL_HAIRCUT_FOLDER_PATH=resource\NeuralHaircut"
SET "PIXIE_PATH=resource\NeuralHaircut\PIXIE\pixie_data.tar.gz" SET "PIXIE_PATH=resource\NeuralHaircut\PIXIE\pixie_data\pixie_data"
SET "MATTE_PATH=resource\Matte-Anything\model.pth" SET "MATTE_PATH=resource\Matte-Anything\model.pth"
SET "SAM_PATH=resource\Matte-Anything\sam_vit_h_4b8939.pth" SET "SAM_PATH=resource\Matte-Anything\sam_vit_h_4b8939.pth"
SET "DINO_PATH=resource\Matte-Anything\groundingdino_swint_ogc.pth" SET "DINO_PATH=resource\Matte-Anything\groundingdino_swint_ogc.pth"
@ -98,7 +98,6 @@ REM 创建resource目录结构
mkdir resource 2>nul mkdir resource 2>nul
cd resource cd resource
mkdir NeuralHaircut 2>nul mkdir NeuralHaircut 2>nul
mkdir NeuralHaircut\pretrained_models 2>nul
mkdir Matte-Anything 2>nul mkdir Matte-Anything 2>nul
mkdir openpose 2>nul mkdir openpose 2>nul
mkdir hyperIQA 2>nul mkdir hyperIQA 2>nul
@ -106,12 +105,12 @@ mkdir hyperIQA 2>nul
REM 验证已有文件 REM 验证已有文件
echo 正在检查已有资源文件... echo 正在检查已有资源文件...
SET MISSING_FILES=0 SET MISSING_FILES=0
SET /A TOTAL_FILES=8 SET /A TOTAL_FILES=7
SET /A EXISTING_FILES=0 SET /A EXISTING_FILES=0
SET MISSING_LIST= SET MISSING_LIST=
REM Neural Haircut Diffusion Prior 检查 REM Neural Haircut Diffusion Prior 检查
IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\diffusion_prior\dif_ckpt.pt" ( IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\dif_ckpt.pt" (
echo [√] Neural Haircut Diffusion Prior已存在 echo [√] Neural Haircut Diffusion Prior已存在
SET /A EXISTING_FILES+=1 SET /A EXISTING_FILES+=1
) ELSE ( ) ELSE (
@ -121,7 +120,7 @@ IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\diffusion_prior\dif_ckpt.pt" (
) )
REM Neural Haircut Strand Prior 检查 REM Neural Haircut Strand Prior 检查
IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\strand_prior\strand_ckpt.pt" ( IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\strand_ckpt.pt" (
echo [√] Neural Haircut Strand Prior已存在 echo [√] Neural Haircut Strand Prior已存在
SET /A EXISTING_FILES+=1 SET /A EXISTING_FILES+=1
) ELSE ( ) ELSE (
@ -208,6 +207,7 @@ echo 正在下载Neural Haircut预训练文件夹...
python -m gdown --folder "%NEURAL_HAIRCUT_FOLDER_URL%" python -m gdown --folder "%NEURAL_HAIRCUT_FOLDER_URL%"
REM 更新 diffusion_prior 模型 REM 更新 diffusion_prior 模型
mkdir pretrained_models\diffusion_prior 2>nul
cd pretrained_models\diffusion_prior cd pretrained_models\diffusion_prior
echo 正在更新Neural Haircut Diffusion Prior模型... echo 正在更新Neural Haircut Diffusion Prior模型...
python -m gdown "%NEURAL_HAIRCUT_MODEL_URL%" python -m gdown "%NEURAL_HAIRCUT_MODEL_URL%"
@ -263,14 +263,14 @@ echo 正在验证下载文件...
cd %PROJECT_DIR% cd %PROJECT_DIR%
SET VERIFY_FAILED=0 SET VERIFY_FAILED=0
IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\diffusion_prior\dif_ckpt.pt" ( IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\dif_ckpt.pt" (
echo [×] 缺少Neural Haircut Diffusion Prior echo [×] 缺少Neural Haircut Diffusion Prior
SET /A VERIFY_FAILED+=1 SET /A VERIFY_FAILED+=1
) ELSE ( ) ELSE (
echo [√] Neural Haircut Diffusion Prior验证成功 echo [√] Neural Haircut Diffusion Prior验证成功
) )
IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\strand_prior\strand_ckpt.pt" ( IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\strand_ckpt.pt" (
echo [×] 缺少Neural Haircut Strand Prior echo [×] 缺少Neural Haircut Strand Prior
SET /A VERIFY_FAILED+=1 SET /A VERIFY_FAILED+=1
) ELSE ( ) ELSE (

13
install.bat
View File

@ -164,11 +164,10 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
echo. echo.
echo resource/ echo resource/
echo ├── NeuralHaircut/ echo ├── NeuralHaircut/
echo │ ├── pretrained_models/ echo │ ├── diffusion_prior/
echo │ │ ├── diffusion_prior/ echo │ │ └── model.pt
echo │ │ │ └── dif_ckpt.pt echo │ ├── strand_prior/
echo │ │ └── strand_prior/ echo │ │ └── model.pt
echo │ │ └── strand_ckpt.pt
echo │ └── PIXIE/ echo │ └── PIXIE/
echo │ └── pixie_data/ echo │ └── pixie_data/
echo ├── Matte-Anything/ echo ├── Matte-Anything/
@ -187,13 +186,13 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
) )
REM 检查模型文件是否存在 REM 检查模型文件是否存在
IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior\dif_ckpt.pt" ( IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\diffusion_prior\model.pt" (
echo ERROR: Neural Haircut Diffusion Prior model not found echo ERROR: Neural Haircut Diffusion Prior model not found
echo Please run download_resource.bat first echo Please run download_resource.bat first
exit /b 1 exit /b 1
) )
IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\strand_prior\strand_ckpt.pt" ( IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\strand_prior\model.pt" (
echo ERROR: Neural Haircut Strand Prior model not found echo ERROR: Neural Haircut Strand Prior model not found
echo Please run download_resource.bat first echo Please run download_resource.bat first
exit /b 1 exit /b 1