Update

install.bat

@@ -30,21 +30,36 @@ cd %PROJECT_DIR%\ext && git clone --recursive https://github.com/NVIDIAGameWorks
 cd %PROJECT_DIR%\ext\kaolin && git checkout v0.15.0
 cd %PROJECT_DIR%\ext && git clone https://github.com/SSL92/hyperIQA
 
-REM 安装环境
-%MICROMAMBA_EXE% create -y -p %MAMBA_ROOT_PREFIX%\envs\gaussian_splatting_hair ^
-    python=3.9 pip=23.3.1 setuptools=69.5.1 ^
-    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 conda-forge -c defaults -c anaconda -c fvcore -c iopath -c bottler -c nvidia
+REM 创建主环境
+%MICROMAMBA_EXE% create -y -p %MAMBA_ROOT_PREFIX%\envs\gaussian_splatting_hair python=3.9
+CALL %MICROMAMBA_EXE% activate -p %MAMBA_ROOT_PREFIX%\envs\gaussian_splatting_hair
+pip install -r requirements.txt
+CALL %MICROMAMBA_EXE% deactivate
+
+REM 创建 Matte-Anything 环境
+%MICROMAMBA_EXE% create -y -p %MAMBA_ROOT_PREFIX%\envs\matte_anything python=3.9
+CALL %MICROMAMBA_EXE% activate -p %MAMBA_ROOT_PREFIX%\envs\matte_anything
+pip install -r requirements_matte.txt
+CALL %MICROMAMBA_EXE% deactivate
+
+REM 创建 PIXIE 环境
+%MICROMAMBA_EXE% create -y -p %MAMBA_ROOT_PREFIX%\envs\pixie-env python=3.8
+CALL %MICROMAMBA_EXE% activate -p %MAMBA_ROOT_PREFIX%\envs\pixie-env
+pip install -r requirements_pixie.txt
+CALL %MICROMAMBA_EXE% deactivate
+
+REM 创建 OpenPose 环境
+%MICROMAMBA_EXE% create -y -p %MAMBA_ROOT_PREFIX%\envs\openpose python=3.9
+CALL %MICROMAMBA_EXE% activate -p %MAMBA_ROOT_PREFIX%\envs\openpose
+pip install -r requirements_openpose.txt
+CALL %MICROMAMBA_EXE% deactivate
 
 REM 从resource文件夹拷贝Neural Haircut文件
 xcopy /E /I /Y %PROJECT_DIR%\resource\NeuralHaircut %PROJECT_DIR%\ext\NeuralHaircut
 xcopy /E /I /Y %PROJECT_DIR%\resource\hyperIQA\pretrained %PROJECT_DIR%\ext\hyperIQA\pretrained
 cd %PROJECT_DIR%
-REM 退出gaussian_splatting_hair环境
+
+REM 退出环境
 CALL %MICROMAMBA_EXE% deactivate
 
 REM Matte-Anything

requirements.txt

@@ -1,38 +1,40 @@
-numpy>=1.21.0,<1.25.0
-scipy>=1.7.0,<2.0.0
-pillow>=9.5.0
-tqdm>=4.62.2
-matplotlib>=3.4.2
-opencv-python>=4.5.3
-scikit-image>=0.18.2
-imageio>=2.9.0
-lpips>=0.1.4
-gdown>=4.7.1
-face-alignment==1.3.5
-pytorch3d>=0.7.0,<0.8.0
-ninja>=1.10.2
-trimesh>=3.9.35
-pyhocon>=0.3.59
-wandb>=0.12.0
-tensorboard>=2.6.0
-open3d>=0.13.0
-kaolin>=0.13.0,<0.16.0
-pytorch-lightning>=1.6.0
-plyfile>=0.8.1
-icecream>=2.1.3
-einops>=0.6.0
-accelerate>=0.18.0
-jsonmerge>=1.9.0
-easydict>=1.9
-tensorboardx>=2.6
-fvcore>=0.1.5
-toml>=0.10.2
-pysdf>=0.1.9
-clean-fid>=0.1.35
-clip>=0.2.0
-torchdiffeq>=0.2.3
-torchsde>=0.2.5
-resize-right>=0.0.2
-colmap>=3.10
-torch>=1.12.0,<2.0.0
-torchvision>=0.13.0,<1.0.0 
+# Core dependencies from environment.yml
+python==3.9
+pip==23.3.1
+setuptools==69.5.1
+plyfile==0.8.1
+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
+
+# Pip packages
+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
+
+# Local packages
+-e ./ext/pytorch3d
+-e ./ext/NeuralHaircut/npbgpp
+-e ./ext/simple-knn
+-e ./ext/diff_gaussian_rasterization_hair
+-e ./ext/kaolin 

requirements_matte.txt

@@ -0,0 +1,18 @@
+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
+
+# Git packages
+git+https://github.com/facebookresearch/segment-anything.git
+git+https://github.com/facebookresearch/detectron2.git
+supervision==0.22.0 

requirements_openpose.txt

@@ -0,0 +1 @@
+cmake==3.20 

requirements_pixie.txt

@@ -0,0 +1,13 @@
+python==3.8
+pytorch==2.0.0
+torchvision==0.15.0
+torchaudio==2.0.0
+pytorch-cuda==11.8
+fvcore
+pytorch3d==0.7.5
+kornia
+matplotlib
+pyyaml==5.4.1
+
+# Git packages
+git+https://github.com/1adrianb/face-alignment.git@54623537fd9618ca7c15688fd85aba706ad92b59