From 36fd2b5a853318ef53f92d5593fde9bc7073e9f1 Mon Sep 17 00:00:00 2001 From: Jeffreytsai1004 Date: Thu, 20 Feb 2025 03:59:54 +0800 Subject: [PATCH] Update --- download_resource.bat | 6 ++ fetch_model.bat | 13 ++-- install.bat | 139 ++++++++++++++++++++---------------------- 3 files changed, 76 insertions(+), 82 deletions(-) diff --git a/download_resource.bat b/download_resource.bat index f1c9169..c22e1ad 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -40,6 +40,12 @@ echo "Downloading openpose..." @CALL tar -xvzf models.tar.gz @CALL cd ..\.. +echo "Downloading SMPL-X..." +@CALL cd resource\PIXIE\data +@CALL python -m gdown 1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW +@CALL cd ..\.. + + echo "Download complete" diff --git a/fetch_model.bat b/fetch_model.bat index 1aff395..e6f307b 100644 --- a/fetch_model.bat +++ b/fetch_model.bat @@ -4,16 +4,16 @@ mkdir .\resource\PIXIE\data echo. echo You need to register at https://smpl-x.is.tue.mpg.de -set /P username_smplx="Username (SMPL-X): " -set /P password_smplx="Password (SMPL-X): " +set /P username_smplx="jeffreytsai1004@gmail.com " +set /P password_smplx="XXXXXX" 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 echo. echo You need to register at https://pixie.is.tue.mpg.de/ -set /P username_pixie="Username (PIXIE): " -set /P password_pixie="Password (PIXIE): " +set /P username_pixie="jeffreytsai1004@gmail.com " +set /P password_pixie="XXXXXX" 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 @@ -21,11 +21,6 @@ curl -o ".\resource\PIXIE\data\pixie_model.tar" --post-fields "username=%usernam 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 -cd data -echo Extracting utilities.zip... -powershell -Command "Expand-Archive -Path 'utilities.zip' -DestinationPath '.'" -echo Extraction complete. - echo. echo Model download and extraction completed. pause \ No newline at end of file diff --git a/install.bat b/install.bat index 6abafb6..1416f7b 100644 --- a/install.bat +++ b/install.bat @@ -8,105 +8,98 @@ @CALL SET VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat" @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 python -m pip install pip --upgrade -@CALL python -m pip install gdown==5.2.0 -@CALL python -m pip install gcc==10.4.0 -@CALL python -m pip install -r requirements.txt +@CALL python -m pip install --upgrade pip +@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 cd %PROJECT_DIR%\ext && git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose --depth 1 -@CALL cd %PROJECT_DIR%\ext\openpose && git submodule update --init --recursive --remote -@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/hustvl/Matte-Anything -@CALL cd %PROJECT_DIR%\ext\Matte-Anything && git clone https://github.com/IDEA-Research/GroundingDINO.git -@CALL cd %PROJECT_DIR%\ext && git clone git@github.com:egorzakharov/NeuralHaircut.git --recursive -@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/facebookresearch/pytorch3d -@CALL cd %PROJECT_DIR%\ext\pytorch3d && git checkout 2f11ddc5ee7d6bd56f2fb6744a16776fab6536f7 -@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/camenduru/simple-knn -@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/g-truc/glm -@CALL cd %PROJECT_DIR%\ext\diff_gaussian_rasterization_hair\third_party\glm && git checkout 5c46b9c07008ae65cb81ab79cd677ecc1934b903 -@CALL cd %PROJECT_DIR%\ext && git clone --recursive https://github.com/NVIDIAGameWorks/kaolin -@CALL cd %PROJECT_DIR%\ext\kaolin && git checkout v0.15.0 -@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/SSL92/hyperIQA -@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/yfeng95/PIXIE +REM Clone repositories +@CALL cd %PROJECT_DIR% +@CALL mkdir ext +@CALL cd ext +@CALL git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose --depth 1 +@CALL cd openpose && git submodule update --init --recursive --remote +@CALL cd .. +@CALL git clone https://github.com/hustvl/Matte-Anything +@CALL cd Matte-Anything && git clone https://github.com/IDEA-Research/GroundingDINO.git +@CALL cd .. +@CALL git clone https://github.com/egorzakharov/NeuralHaircut.git --recursive +@CALL git clone https://github.com/facebookresearch/pytorch3d +@CALL cd pytorch3d && git checkout 2f11ddc5ee7d6bd56f2fb6744a16776fab6536f7 +@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\ -@CALL copy %PROJECT_DIR%\resource\hyperIQA\* %PROJECT_DIR%\ext\hyperIQA\ -@CALL copy %PROJECT_DIR%\resource\openpose\* %PROJECT_DIR%\ext\openpose\ -@CALL copy %PROJECT_DIR%\resource\PIXIE\data\* %PROJECT_DIR%\ext\PIXIE\data\ -@CALL copy %PROJECT_DIR%\resource\NeuralHaircut\PIXIE\data\* %PROJECT_DIR%\ext\NeuralHaircut\PIXIE\data\ -@CALL copy %PROJECT_DIR%\resource\Matte-Anything\pretrained\* %PROJECT_DIR%\ext\Matte-Anything\pretrained\ -@CALL %PROJECT_DIR%\ext\NeuralHaircut\PIXIE\data\ +REM Copy resources +@CALL cd %PROJECT_DIR% +@CALL copy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\*" "%PROJECT_DIR%\ext\NeuralHaircut\" +@CALL copy /E /I /Y "%PROJECT_DIR%\resource\hyperIQA\*" "%PROJECT_DIR%\ext\hyperIQA\" +@CALL copy /E /I /Y "%PROJECT_DIR%\resource\openpose\*" "%PROJECT_DIR%\ext\openpose\" +@CALL copy /E /I /Y "%PROJECT_DIR%\resource\PIXIE\data\*" "%PROJECT_DIR%\ext\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 cd %PROJECT_DIR% -@CALL "%~dp0condabin\micromamba.bat" deactivate gaussian_splatting_hair +@CALL "%~dp0condabin\micromamba.bat" deactivate -@CALL "%~dp0micromamba.exe" create -n matte_anything ^ - python=3.9 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 ^ +REM Create matte_anything environment +@CALL "%~dp0micromamba.exe" create -n matte_anything python=3.9 ^ + 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 + @CALL "%~dp0condabin\micromamba.bat" activate matte_anything +@CALL python -m pip install fairscale wget @CALL cd %PROJECT_DIR%\ext\Matte-Anything @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 %VS_VCVARS% && python setup.py build develop -@CALL cd %PROJECT_DIR%\ext\Matte-Anything\GroundingDINO && pip install -e . +@CALL CALL %VS_VCVARS% && python setup.py build develop +@CALL cd %PROJECT_DIR%\ext\Matte-Anything\GroundingDINO +@CALL pip install -e . @CALL pip install supervision==0.22.0 -@CALL cd %PROJECT_DIR%\ext\Matte-Anything && mkdir pretrained @CALL cd %PROJECT_DIR% @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 "%~dp0condabin\micromamba.bat" activate openpose @CALL cd %PROJECT_DIR%\ext\openpose @CALL mkdir build @CALL cd build -@CALL %VS_VCVARS% -@CALL cmake .. -G "Visual Studio 17 2022" -A x64 -T host=x64 -DBUILD_PYTHON=true -DUSE_CUDNN=off +@CALL CALL %VS_VCVARS% +@CALL cmake .. -G "Visual Studio 16 2019" -A x64 -T host=x64 -DBUILD_PYTHON=true -DUSE_CUDNN=off @CALL cmake --build . --config Release @CALL cd %PROJECT_DIR% @CALL "%~dp0condabin\micromamba.bat" deactivate -@CALL "%~dp0micromamba.exe" create -n pixie-env python=3.8 pytorch==2.0.0 torchvision torchaudio ^ - pytorch-cuda=11.8 fvcore pytorch3d==0.7.5 kornia matplotlib ^ - -c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -r "%~dp0\" -y +REM Create pixie-env environment +@CALL "%~dp0micromamba.exe" create -n pixie-env python=3.8 ^ + 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 pip install pytorch3d==0.7.5 @CALL cd %PROJECT_DIR%\ext\PIXIE @CALL pip install pyyaml==5.4.1 @CALL pip install git+https://github.com/1adrianb/face-alignment.git@54623537fd9618ca7c15688fd85aba706ad92b59 -@CALL mkdir data -@CALL 7z x utilities.zip -@CALL tar -xf pixie_data.tar -@CALL "%~dp0condabin\micromamba.bat" deactivate - - - - - - - - - - - - - - - - - - - - - - - - - - - - - +@CALL cd %PROJECT_DIR% +@CALL "%~dp0condabin\micromamba.bat" deactivate \ No newline at end of file