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:33:54 +08:00
parent f31b034c60
commit f0e6d9c90d
2 changed files with 17 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
download_resource.bat
View File

@ -39,7 +39,7 @@ SET "MODEL_DINO_PATH=resource\Matte-Anything\groundingdino_swint_ogc.pth"
SET "MODEL_DINO_URL=https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth" SET "MODEL_DINO_URL=https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth"
SET "MODEL_OPENPOSE_NAME=OpenPose模型" SET "MODEL_OPENPOSE_NAME=OpenPose模型"
SET "MODEL_OPENPOSE_PATH=resource\openpose\models\pose_iter_584000.caffemodel" SET "MODEL_OPENPOSE_PATH=resource\openpose\models\pose\coco\pose_iter_584000.caffemodel"
SET "MODEL_OPENPOSE_URL=1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" SET "MODEL_OPENPOSE_URL=1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV"
SET "MODEL_HYPERIQA_NAME=hyperIQA模型" SET "MODEL_HYPERIQA_NAME=hyperIQA模型"
@ -115,7 +115,7 @@ cd resource
mkdir NeuralHaircut\pretrained_models\diffusion_prior 2>nul mkdir NeuralHaircut\pretrained_models\diffusion_prior 2>nul
mkdir PIXIE 2>nul mkdir PIXIE 2>nul
mkdir Matte-Anything 2>nul mkdir Matte-Anything 2>nul
mkdir openpose\models 2>nul mkdir openpose\models\pose\coco 2>nul
mkdir hyperIQA\pretrained 2>nul mkdir hyperIQA\pretrained 2>nul
REM 验证已有文件 REM 验证已有文件
@ -257,7 +257,7 @@ IF EXIST "%MODEL_MATTE_PATH%" (
) )
REM 下载 OpenPose 模型 REM 下载 OpenPose 模型
cd ..\openpose\models cd ..\openpose\models\pose\coco
echo 正在下载OpenPose模型... echo 正在下载OpenPose模型...
IF EXIST "%MODEL_OPENPOSE_PATH%" ( IF EXIST "%MODEL_OPENPOSE_PATH%" (
echo %MODEL_OPENPOSE_NAME%已存在，跳过下载... echo %MODEL_OPENPOSE_NAME%已存在，跳过下载...

23
install.bat
View File

@ -127,14 +127,10 @@ git clone https://github.com/SSL92/hyperIQA
cd .. cd ..
REM 创建环境 REM 创建环境
CALL "%MICROMAMBA_EXE%" create -p %MAMBA_ROOT_PREFIX%\envs\gaussian_splatting_hair python=3.8 pytorch=2.0.0 torchvision pytorch-cuda=11.8 cmake ninja setuptools=58.2.0 -c pytorch -c nvidia -c conda-forge -y CALL "%MICROMAMBA_EXE%" env create -f environment.yml
CALL "%MICROMAMBA_EXE%" create -p %MAMBA_ROOT_PREFIX%\envs\matte_anything pytorch=2.0.0 pytorch-cuda=11.8 torchvision tensorboard timm=0.5.4 opencv=4.5.3 mkl=2024.0 setuptools=58.2.0 easydict wget scikit-image gradio=3.46.1 fairscale supervision==0.22.0 -c pytorch -c nvidia -c conda-forge -y
CALL "%MICROMAMBA_EXE%" create -p %MAMBA_ROOT_PREFIX%\envs\openpose python=3.8 cmake=3.20 -c conda-forge -y
CALL "%MICROMAMBA_EXE%" create -p %MAMBA_ROOT_PREFIX%\envs\pixie-env python=3.8 pytorch=2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 fvcore pytorch3d==0.7.5 kornia matplotlib -c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -c bottler -c iopath -y
REM 安装 gaussian_splatting_hair 环境 REM 安装 gaussian_splatting_hair 环境
CALL "%MICROMAMBA_EXE%" activate -p %MAMBA_ROOT_PREFIX%\envs\gaussian_splatting_hair CALL "%MICROMAMBA_EXE%" activate -p %MAMBA_ROOT_PREFIX%\envs\gaussian_splatting_hair
pip install -r requirements.txt
REM 安装外部库 REM 安装外部库
cd %PROJECT_DIR%\ext\pytorch3d cd %PROJECT_DIR%\ext\pytorch3d
@ -156,14 +152,23 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
echo. echo.
echo resource/ echo resource/
echo ├── NeuralHaircut/ echo ├── NeuralHaircut/
echo │ ├── diffusion_prior/ echo │ ├── pretrained_models/
echo │ └── PIXIE/ echo │ │ └── diffusion_prior/
echo │ │ └── model.pt
echo ├── PIXIE/
echo │ └── pixie_data/
echo ├── Matte-Anything/ echo ├── Matte-Anything/
echo │ ├── sam_vit_h_4b8939.pth echo │ ├── sam_vit_h_4b8939.pth
echo │ ├── groundingdino_swint_ogc.pth echo │ ├── groundingdino_swint_ogc.pth
echo │ └── model.pth echo │ └── model.pth
echo └── openpose/ echo ├── openpose/
echo └── models/ echo │ └── models/
echo │ └── pose/
echo │ └── coco/
echo │ └── pose_iter_584000.caffemodel
echo └── hyperIQA/
echo └── pretrained/
echo └── hyperIQA.pth
exit /b 1 exit /b 1
) )