diff --git a/download_resource.bat b/download_resource.bat index e3f4e6a..a5b65eb 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -70,19 +70,9 @@ echo "Downloading PIXIE model..." @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=utilities.zip&resume=1" -O utilities.zip --no-check-certificate --continue -@CALL tar -xf pixie_model.tar -@CALL unzip -o utilities.zip +@CALL tar -xvzf utilities.zip @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" if not exist "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior\dif_ckpt.pth" ( echo ERROR: Failed to download NeuralHaircut models @@ -119,6 +109,11 @@ if not exist "%PROJECT_DIR%\resource\PIXIE\data\utilities" ( 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" pause diff --git a/install.bat b/install.bat index 81820a3..de8bb2f 100644 --- a/install.bat +++ b/install.bat @@ -91,9 +91,7 @@ REM Create openpose environment @CALL "%~dp0micromamba.exe" create -n openpose cmake=3.20 -c conda-forge -r "%~dp0\" -y @CALL "%~dp0condabin\micromamba.bat" activate openpose @CALL cd %PROJECT_DIR%\ext\openpose -@CALL python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV -@CALL tar -xvzf models.tar.gz -@CALL del /f /q models.tar.gz +@CALL copy %PROJECT_DIR%\resource\openpose\models.tar.gz %PROJECT_DIR%\ext\openpose\ @CALL mkdir build @CALL cd build @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 ^ -c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -r "%~dp0\" -y @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 pyyaml==5.4.1 @CALL pip install git+https://github.com/1adrianb/face-alignment.git@54623537fd9618ca7c15688fd85aba706ad92b59