105 lines
4.9 KiB
Batchfile
105 lines
4.9 KiB
Batchfile
@echo off
|
|
|
|
@CALL "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\"
|
|
@CALL SET PROJECT_DIR=%~dp0
|
|
@CALL SET CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\"
|
|
@CALL SET BLENDER_DIR="C:\Program Files\Blender Foundation\Blender 3.6\"
|
|
@CALL SET VS_DIR="D:\Program Files\Microsoft Visual Studio\2022\Community\"
|
|
@CALL SET VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat"
|
|
@CALL SET PATH=%CUDA_HOME%\bin;%PROJECT_DIR%\condabin;%PATH%
|
|
|
|
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 --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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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/facebookresearch/detectron2
|
|
@CALL cd detectron2
|
|
@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%
|
|
|
|
@CALL "%~dp0condabin\micromamba.bat" deactivate
|
|
|
|
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 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
|
|
|
|
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 cd %PROJECT_DIR%
|
|
@CALL "%~dp0condabin\micromamba.bat" deactivate |