From e3a4fd4645c31de70fa069549770cbf9b3d31899 Mon Sep 17 00:00:00 2001 From: Jeffreytsai1004 Date: Tue, 18 Feb 2025 00:54:47 +0800 Subject: [PATCH] Update --- install.bat | 35 +++++++++++++----- requirements.txt | 78 ++++++++++++++++++++------------------- requirements_matte.txt | 18 +++++++++ requirements_openpose.txt | 1 + requirements_pixie.txt | 13 +++++++ 5 files changed, 97 insertions(+), 48 deletions(-) create mode 100644 requirements_matte.txt create mode 100644 requirements_openpose.txt create mode 100644 requirements_pixie.txt diff --git a/install.bat b/install.bat index 66f5874..e7a69be 100644 --- a/install.bat +++ b/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 diff --git a/requirements.txt b/requirements.txt index 9a51046..e08742b 100644 --- a/requirements.txt +++ b/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 \ No newline at end of file +# 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 \ No newline at end of file diff --git a/requirements_matte.txt b/requirements_matte.txt new file mode 100644 index 0000000..b3fda22 --- /dev/null +++ b/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 \ No newline at end of file diff --git a/requirements_openpose.txt b/requirements_openpose.txt new file mode 100644 index 0000000..e1eb5a9 --- /dev/null +++ b/requirements_openpose.txt @@ -0,0 +1 @@ +cmake==3.20 \ No newline at end of file diff --git a/requirements_pixie.txt b/requirements_pixie.txt new file mode 100644 index 0000000..5c59ee0 --- /dev/null +++ b/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 \ No newline at end of file