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-20 03:59:54 +08:00
parent 7d31e68700
commit 36fd2b5a85
3 changed files with 76 additions and 82 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
download_resource.bat
View File

@ -40,6 +40,12 @@ echo "Downloading openpose..."
@CALL tar -xvzf models.tar.gz @CALL tar -xvzf models.tar.gz
@CALL cd ..\.. @CALL cd ..\..
echo "Downloading SMPL-X..."
@CALL cd resource\PIXIE\data
@CALL python -m gdown 1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW
@CALL cd ..\..
echo "Download complete" echo "Download complete"

13
fetch_model.bat
View File

@ -4,16 +4,16 @@ mkdir .\resource\PIXIE\data
echo. echo.
echo You need to register at https://smpl-x.is.tue.mpg.de echo You need to register at https://smpl-x.is.tue.mpg.de
set /P username_smplx="Username (SMPL-X): " set /P username_smplx="jeffreytsai1004@gmail.com "
set /P password_smplx="Password (SMPL-X): " set /P password_smplx="XXXXXX"
echo Downloading SMPL-X 2020 (neutral SMPL-X model)... echo Downloading SMPL-X 2020 (neutral SMPL-X model)...
curl -o ".\resource\PIXIE\data\SMPLX_NEUTRAL_2020.npz" --post-fields "username=%username_smplx%&password=%password_smplx%" "https://download.is.tue.mpg.de/download.php?domain=smplx&sfile=SMPLX_NEUTRAL_2020.npz&resume=1" --insecure --continue curl -o ".\resource\PIXIE\data\SMPLX_NEUTRAL_2020.npz" --post-fields "username=%username_smplx%&password=%password_smplx%" "https://download.is.tue.mpg.de/download.php?domain=smplx&sfile=SMPLX_NEUTRAL_2020.npz&resume=1" --insecure --continue
echo. echo.
echo You need to register at https://pixie.is.tue.mpg.de/ echo You need to register at https://pixie.is.tue.mpg.de/
set /P username_pixie="Username (PIXIE): " set /P username_pixie="jeffreytsai1004@gmail.com "
set /P password_pixie="Password (PIXIE): " set /P password_pixie="XXXXXX"
echo Downloading PIXIE pretrained model... echo Downloading PIXIE pretrained model...
curl -o ".\resource\PIXIE\data\pixie_model.tar" --post-fields "username=%username_pixie%&password=%password_pixie%" "https://download.is.tue.mpg.de/download.php?domain=pixie&sfile=pixie_model.tar&resume=1" --insecure --continue curl -o ".\resource\PIXIE\data\pixie_model.tar" --post-fields "username=%username_pixie%&password=%password_pixie%" "https://download.is.tue.mpg.de/download.php?domain=pixie&sfile=pixie_model.tar&resume=1" --insecure --continue
@ -21,11 +21,6 @@ curl -o ".\resource\PIXIE\data\pixie_model.tar" --post-fields "username=%usernam
echo Downloading PIXIE utilities... echo Downloading PIXIE utilities...
curl -o ".\resource\PIXIE\data\utilities.zip" --post-fields "username=%username_pixie%&password=%password_pixie%" "https://download.is.tue.mpg.de/download.php?domain=pixie&sfile=utilities.zip&resume=1" --insecure --continue curl -o ".\resource\PIXIE\data\utilities.zip" --post-fields "username=%username_pixie%&password=%password_pixie%" "https://download.is.tue.mpg.de/download.php?domain=pixie&sfile=utilities.zip&resume=1" --insecure --continue
cd data
echo Extracting utilities.zip...
powershell -Command "Expand-Archive -Path 'utilities.zip' -DestinationPath '.'"
echo Extraction complete.
echo. echo.
echo Model download and extraction completed. echo Model download and extraction completed.
pause pause

137
install.bat
View File

@ -8,105 +8,98 @@
@CALL SET VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat" @CALL SET VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat"
@CALL SET PATH=%CUDA_HOME%\bin;%PROJECT_DIR%\condabin;%PATH% @CALL SET PATH=%CUDA_HOME%\bin;%PROJECT_DIR%\condabin;%PATH%
@CALL "%~dp0micromamba.exe" create -n gaussian_splatting_hair python=3.9 -c pytorch -c nvidia -c conda-forge -c anaconda -c fvcore -c iopath -c bottler -c nvidia -r "%~dp0\" -y REM Create gaussian_splatting_hair environment
@CALL "%~dp0micromamba.exe" create -n gaussian_splatting_hair python=3.9 ^
pytorch=2.1.1 torchvision=0.16.1 torchaudio=2.1.1 pytorch-cuda=11.8 ^
cmake=3.28.0 pyhocon=0.3.60 icecream=2.1.3 einops=0.6.0 accelerate=0.18.0 ^
jsonmerge=1.9.0 easydict=1.9 iopath=0.1.10 tensorboardx=2.6 scikit-image=0.20.0 ^
fvcore=0.1.5 toml=0.10.2 tqdm=4.66.5 gdown=5.2.0 colmap=3.10 ^
-c pytorch -c nvidia -c conda-forge -c anaconda -c fvcore -c iopath -c bottler -r "%~dp0\" -y
@CALL "%~dp0condabin\micromamba.bat" activate gaussian_splatting_hair @CALL "%~dp0condabin\micromamba.bat" activate gaussian_splatting_hair
@CALL python -m pip install pip --upgrade @CALL python -m pip install --upgrade pip
@CALL python -m pip install gdown==5.2.0 @CALL python -m pip install pysdf==0.1.9 clean-fid==0.1.35 face-alignment==1.4.1 clip==0.2.0 torchdiffeq==0.2.3 torchsde==0.2.5 resize-right==0.0.2
@CALL python -m pip install gcc==10.4.0
@CALL python -m pip install -r requirements.txt
@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose --depth 1 REM Clone repositories
@CALL cd %PROJECT_DIR%\ext\openpose && git submodule update --init --recursive --remote @CALL cd %PROJECT_DIR%
@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/hustvl/Matte-Anything @CALL mkdir ext
@CALL cd %PROJECT_DIR%\ext\Matte-Anything && git clone https://github.com/IDEA-Research/GroundingDINO.git @CALL cd ext
@CALL cd %PROJECT_DIR%\ext && git clone git@github.com:egorzakharov/NeuralHaircut.git --recursive @CALL git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose --depth 1
@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/facebookresearch/pytorch3d @CALL cd openpose && git submodule update --init --recursive --remote
@CALL cd %PROJECT_DIR%\ext\pytorch3d && git checkout 2f11ddc5ee7d6bd56f2fb6744a16776fab6536f7 @CALL cd ..
@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/camenduru/simple-knn @CALL git clone https://github.com/hustvl/Matte-Anything
@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/g-truc/glm @CALL cd Matte-Anything && git clone https://github.com/IDEA-Research/GroundingDINO.git
@CALL cd %PROJECT_DIR%\ext\diff_gaussian_rasterization_hair\third_party\glm && git checkout 5c46b9c07008ae65cb81ab79cd677ecc1934b903 @CALL cd ..
@CALL cd %PROJECT_DIR%\ext && git clone --recursive https://github.com/NVIDIAGameWorks/kaolin @CALL git clone https://github.com/egorzakharov/NeuralHaircut.git --recursive
@CALL cd %PROJECT_DIR%\ext\kaolin && git checkout v0.15.0 @CALL git clone https://github.com/facebookresearch/pytorch3d
@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/SSL92/hyperIQA @CALL cd pytorch3d && git checkout 2f11ddc5ee7d6bd56f2fb6744a16776fab6536f7
@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/yfeng95/PIXIE @CALL cd ..
@CALL git clone https://github.com/camenduru/simple-knn
@CALL git clone https://github.com/g-truc/glm
@CALL cd glm && git checkout 5c46b9c07008ae65cb81ab79cd677ecc1934b903
@CALL cd ..
@CALL git clone --recursive https://github.com/NVIDIAGameWorks/kaolin
@CALL cd kaolin && git checkout v0.15.0
@CALL cd ..
@CALL git clone https://github.com/SSL92/hyperIQA
@CALL git clone https://github.com/yfeng95/PIXIE
@CALL copy %PROJECT_DIR%\resource\NeuralHaircut\* %PROJECT_DIR%\ext\NeuralHaircut\ REM Copy resources
@CALL copy %PROJECT_DIR%\resource\hyperIQA\* %PROJECT_DIR%\ext\hyperIQA\ @CALL cd %PROJECT_DIR%
@CALL copy %PROJECT_DIR%\resource\openpose\* %PROJECT_DIR%\ext\openpose\ @CALL copy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\*" "%PROJECT_DIR%\ext\NeuralHaircut\"
@CALL copy %PROJECT_DIR%\resource\PIXIE\data\* %PROJECT_DIR%\ext\PIXIE\data\ @CALL copy /E /I /Y "%PROJECT_DIR%\resource\hyperIQA\*" "%PROJECT_DIR%\ext\hyperIQA\"
@CALL copy %PROJECT_DIR%\resource\NeuralHaircut\PIXIE\data\* %PROJECT_DIR%\ext\NeuralHaircut\PIXIE\data\ @CALL copy /E /I /Y "%PROJECT_DIR%\resource\openpose\*" "%PROJECT_DIR%\ext\openpose\"
@CALL copy %PROJECT_DIR%\resource\Matte-Anything\pretrained\* %PROJECT_DIR%\ext\Matte-Anything\pretrained\ @CALL copy /E /I /Y "%PROJECT_DIR%\resource\PIXIE\data\*" "%PROJECT_DIR%\ext\PIXIE\data\"
@CALL %PROJECT_DIR%\ext\NeuralHaircut\PIXIE\data\ @CALL copy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\data\*" "%PROJECT_DIR%\ext\NeuralHaircut\PIXIE\data\"
@CALL copy /E /I /Y "%PROJECT_DIR%\resource\Matte-Anything\pretrained\*" "%PROJECT_DIR%\ext\Matte-Anything\pretrained\"
@CALL cd %PROJECT_DIR%\ext\NeuralHaircut\PIXIE\data
@CALL tar -xvzf pixie_data.tar.gz @CALL tar -xvzf pixie_data.tar.gz
@CALL cd %PROJECT_DIR% @CALL cd %PROJECT_DIR%
@CALL "%~dp0condabin\micromamba.bat" deactivate gaussian_splatting_hair @CALL "%~dp0condabin\micromamba.bat" deactivate
@CALL "%~dp0micromamba.exe" create -n matte_anything ^ REM Create matte_anything environment
python=3.9 pytorch=2.0.0 pytorch-cuda=11.8 torchvision tensorboard timm=0.5.4 opencv=4.5.3 ^ @CALL "%~dp0micromamba.exe" create -n matte_anything python=3.9 ^
mkl=2024.0 setuptools=58.2.0 easydict wget scikit-image gradio=3.46.1 fairscale ^ 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 scikit-image gradio=3.46.1 ^
-c pytorch -c nvidia -c conda-forge -r "%~dp0\" -y -c pytorch -c nvidia -c conda-forge -r "%~dp0\" -y
@CALL "%~dp0condabin\micromamba.bat" activate matte_anything @CALL "%~dp0condabin\micromamba.bat" activate matte_anything
@CALL python -m pip install fairscale wget
@CALL cd %PROJECT_DIR%\ext\Matte-Anything @CALL cd %PROJECT_DIR%\ext\Matte-Anything
@CALL pip install git+https://github.com/facebookresearch/segment-anything.git @CALL pip install git+https://github.com/facebookresearch/segment-anything.git
@CALL git clone https://github.com/conansherry/detectron2 %PROJECT_DIR%\ext\Matte-Anything\detectron2 @CALL git clone https://github.com/facebookresearch/detectron2
@CALL cd detectron2 @CALL cd detectron2
@CALL %VS_VCVARS% && python setup.py build develop @CALL CALL %VS_VCVARS% && python setup.py build develop
@CALL cd %PROJECT_DIR%\ext\Matte-Anything\GroundingDINO && pip install -e . @CALL cd %PROJECT_DIR%\ext\Matte-Anything\GroundingDINO
@CALL pip install -e .
@CALL pip install supervision==0.22.0 @CALL pip install supervision==0.22.0
@CALL cd %PROJECT_DIR%\ext\Matte-Anything && mkdir pretrained
@CALL cd %PROJECT_DIR% @CALL cd %PROJECT_DIR%
@CALL "%~dp0condabin\micromamba.bat" deactivate @CALL "%~dp0condabin\micromamba.bat" deactivate
@CALL git submodule update --init --recursive --remote
REM Create openpose environment
@CALL "%~dp0micromamba.exe" create -n openpose cmake=3.20 -c conda-forge -r "%~dp0\" -y @CALL "%~dp0micromamba.exe" create -n openpose cmake=3.20 -c conda-forge -r "%~dp0\" -y
@CALL "%~dp0condabin\micromamba.bat" activate openpose @CALL "%~dp0condabin\micromamba.bat" activate openpose
@CALL cd %PROJECT_DIR%\ext\openpose @CALL cd %PROJECT_DIR%\ext\openpose
@CALL mkdir build @CALL mkdir build
@CALL cd build @CALL cd build
@CALL %VS_VCVARS% @CALL CALL %VS_VCVARS%
@CALL cmake .. -G "Visual Studio 17 2022" -A x64 -T host=x64 -DBUILD_PYTHON=true -DUSE_CUDNN=off @CALL cmake .. -G "Visual Studio 16 2019" -A x64 -T host=x64 -DBUILD_PYTHON=true -DUSE_CUDNN=off
@CALL cmake --build . --config Release @CALL cmake --build . --config Release
@CALL cd %PROJECT_DIR% @CALL cd %PROJECT_DIR%
@CALL "%~dp0condabin\micromamba.bat" deactivate @CALL "%~dp0condabin\micromamba.bat" deactivate
@CALL "%~dp0micromamba.exe" create -n pixie-env python=3.8 pytorch==2.0.0 torchvision torchaudio ^ REM Create pixie-env environment
pytorch-cuda=11.8 fvcore pytorch3d==0.7.5 kornia matplotlib ^ @CALL "%~dp0micromamba.exe" create -n pixie-env python=3.8 ^
-c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -r "%~dp0\" -y pytorch=2.0.0 torchvision torchaudio pytorch-cuda=11.8 fvcore kornia matplotlib ^
-c pytorch -c nvidia -c fvcore -c conda-forge -r "%~dp0\" -y
@CALL "%~dp0condabin\micromamba.bat" activate pixie-env @CALL "%~dp0condabin\micromamba.bat" activate pixie-env
@CALL pip install pytorch3d==0.7.5
@CALL cd %PROJECT_DIR%\ext\PIXIE @CALL cd %PROJECT_DIR%\ext\PIXIE
@CALL pip install pyyaml==5.4.1 @CALL pip install pyyaml==5.4.1
@CALL pip install git+https://github.com/1adrianb/face-alignment.git@54623537fd9618ca7c15688fd85aba706ad92b59 @CALL pip install git+https://github.com/1adrianb/face-alignment.git@54623537fd9618ca7c15688fd85aba706ad92b59
@CALL mkdir data @CALL cd %PROJECT_DIR%
@CALL 7z x utilities.zip
@CALL tar -xf pixie_data.tar
@CALL "%~dp0condabin\micromamba.bat" deactivate @CALL "%~dp0condabin\micromamba.bat" deactivate