update installation

README.md

@@ -29,8 +29,16 @@ Reconstructing 3D clothed human avatars from single images is a challenging task
 - PyTorch = 1.13.0 (official [Get Started](https://pytorch.org/get-started/locally/))
 - PyTorch3D (official [INSTALL.md](https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md), recommend [install-from-local-clone](https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md#2-install-from-a-local-clone))
 
-Our model uses a environment similar to [ICON](https://github.com/YuliangXiu/ICON), please follow the [ICON Installation](https://github.com/YuliangXiu/ICON/blob/master/docs/installation.md) to install the environment.
+```bash
+
+git clone https://github.com/River-Zhang/GTA.git
+sudo apt-get install libeigen3-dev ffmpeg
+cd GTA
+conda env create -f environment.yaml
+conda activate gta
+pip install -r requirements.txt
 
+```
 Please download the [checkpoint](https://drive.google.com/file/d/1zgpW8A0-sc24o-e24RphO-h9qxo-DRth/view?usp=sharing) and place them in ./data/ckpt
 
 Please follow [ICON](https://github.com/YuliangXiu/ICON/blob/master/docs/installation.md) to download the extra data, such as HPS and SMPL.

environment.yaml

@@ -0,0 +1,21 @@
+name: gta
+channels:
+  - pytorch
+  - nvidia
+  - conda-forge
+  - fvcore
+  - iopath
+  - bottler
+  - defaults
+dependencies:
+  - python=3.8
+  - pytorch-cuda=11.6
+  - pytorch=1.13.0
+  - nvidiacub
+  - torchvision
+  - fvcore
+  - iopath
+  - pyembree
+  - cupy
+  - cython
+  - pip

requirements.txt

@@ -0,0 +1,37 @@
+matplotlib==3.5.0
+Pillow>=9.0.1
+PyOpenGL
+PyOpenGL_accelerate
+PyYAML>=6.0
+yacs>=0.1.8
+scikit-image==0.19.1
+termcolor
+tqdm==4.62.3
+loguru==0.5.3
+trimesh==3.9.35
+flatten_dict==0.4.2
+ipykernel==5.3.4
+ipywidgets==7.6.5
+jpeg4py
+shapely==1.7.1
+vedo==2020.4.2
+rtree==0.9.7
+pytorch_lightning==1.2.5
+PyMCubes
+kornia
+chumpy
+tinyobjloader==2.0.0rc7
+cython==0.29.20
+opencv-python
+opencv_contrib_python
+scikit-learn
+simple-romp==1.0.4
+protobuf==3.20.0
+pymeshlab
+dataclasses>=0.6
+numpy==1.22.4
+pymeshlab==2022.2.post4
+git+https://github.com/facebookresearch/[email protected]
+git+https://github.com/YuliangXiu/neural_voxelization_layer.git
+git+https://github.com/NVIDIAGameWorks/kaolin.git
+git+https://github.com/YuliangXiu/rembg.git@hf