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-22 02:43:30 +08:00
parent 70f8444f29
commit ea325f8453
2 changed files with 9 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
download_resource.bat
View File

@ -70,19 +70,9 @@ echo "Downloading PIXIE model..."
@CALL cd resource\PIXIE\data @CALL cd resource\PIXIE\data
@CALL wget --post-data "username=%PIXIE_USER%&password=%PIXIE_PASS%" "https://download.is.tue.mpg.de/download.php?domain=pixie&sfile=pixie_model.tar&resume=1" -O pixie_model.tar --no-check-certificate --continue @CALL wget --post-data "username=%PIXIE_USER%&password=%PIXIE_PASS%" "https://download.is.tue.mpg.de/download.php?domain=pixie&sfile=pixie_model.tar&resume=1" -O pixie_model.tar --no-check-certificate --continue
@CALL wget --post-data "username=%PIXIE_USER%&password=%PIXIE_PASS%" "https://download.is.tue.mpg.de/download.php?domain=pixie&sfile=utilities.zip&resume=1" -O utilities.zip --no-check-certificate --continue @CALL wget --post-data "username=%PIXIE_USER%&password=%PIXIE_PASS%" "https://download.is.tue.mpg.de/download.php?domain=pixie&sfile=utilities.zip&resume=1" -O utilities.zip --no-check-certificate --continue
@CALL tar -xf pixie_model.tar @CALL tar -xvzf utilities.zip
@CALL unzip -o utilities.zip
@CALL cd %PROJECT_DIR% @CALL cd %PROJECT_DIR%
echo "Clean up temporary files"
@CALL cd %PROJECT_DIR%\ext\NeuralHaircut\PIXIE
@CALL del /f /q pixie_data.tar.gz
@CALL cd %PROJECT_DIR%\ext\openpose
@CALL del /f /q models.tar.gz
@CALL cd %PROJECT_DIR%\resource\PIXIE\data
@CALL del /f /q pixie_model.tar
@CALL del /f /q utilities.zip
echo "Check downloads" echo "Check downloads"
if not exist "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior\dif_ckpt.pth" ( if not exist "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior\dif_ckpt.pth" (
echo ERROR: Failed to download NeuralHaircut models echo ERROR: Failed to download NeuralHaircut models
@ -119,6 +109,11 @@ if not exist "%PROJECT_DIR%\resource\PIXIE\data\utilities" (
exit /b 1 exit /b 1
) )
if not exist "%PROJECT_DIR%\resource\PIXIE\data\SMPLX_NEUTRAL_2020.npz" (
echo ERROR: Failed to download SMPLX_NEUTRAL_2020.npz
exit /b 1
)
echo "Download complete" echo "Download complete"
pause pause

8
install.bat
View File

@ -91,9 +91,7 @@ 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 python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV @CALL copy %PROJECT_DIR%\resource\openpose\models.tar.gz %PROJECT_DIR%\ext\openpose\
@CALL tar -xvzf models.tar.gz
@CALL del /f /q models.tar.gz
@CALL mkdir build @CALL mkdir build
@CALL cd build @CALL cd build
@CALL CALL %VS_VCVARS% @CALL CALL %VS_VCVARS%
@ -108,8 +106,8 @@ REM Create pixie-env environment
pytorch=2.0.0 torchvision torchaudio pytorch-cuda=11.8 fvcore kornia matplotlib ^ pytorch=2.0.0 torchvision torchaudio pytorch-cuda=11.8 fvcore kornia matplotlib ^
-c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -r "%~dp0\" -y -c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -r "%~dp0\" -y
@CALL "%~dp0condabin\micromamba.bat" activate pixie-env @CALL "%~dp0condabin\micromamba.bat" activate pixie-env
@CALL copy %PROJECT_DIR%\resource\PIXIE\data\* %PROJECT_DIR%\ext\PIXIE\ @CALL cd %PROJECT_DIR%\ext\PIXIE\
@CALL cd %PROJECT_DIR%\ext\PIXIE @CALL copy %PROJECT_DIR%\resource\PIXIE\data\* %PROJECT_DIR%\ext\PIXIE\data
@CALL pip install pytorch3d==0.7.5 @CALL pip install pytorch3d==0.7.5
@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