Skip to content

Latest commit

 

History

History
46 lines (37 loc) · 1.6 KB

INSTALL.md

File metadata and controls

46 lines (37 loc) · 1.6 KB

Set up the python environment

conda create -n neuralbody python=3.7
conda activate neuralbody

# make sure that the pytorch cuda is consistent with the system cuda
# e.g., if your system cuda is 10.0, install torch 1.4 built from cuda 10.0
pip install torch==1.4.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html

pip install -r requirements.txt

# install spconv
cd
git clone https://github.com/traveller59/spconv --recursive
cd spconv
git checkout abf0acf30f5526ea93e687e3f424f62d9cd8313a
git submodule update --init --recursive
export CUDA_HOME="/usr/local/cuda-10.0"
python setup.py bdist_wheel
cd dist
pip install spconv-1.2.1-cp36-cp36m-linux_x86_64.whl

Set up datasets

People-Snapshot dataset

  1. Download the People-Snapshot dataset here.
  2. Process the People-Snapshot dataset using the script.
  3. Create a soft link:
    ROOT=/path/to/neuralbody
    cd $ROOT/data
    ln -s /path/to/people_snapshot people_snapshot
    

ZJU-Mocap dataset

  1. If someone wants to download the ZJU-Mocap dataset, please fill the form to obtain the download link. Another way is filling in the agreement and emailing me ([email protected]) and cc Xiaowei Zhou ([email protected]) to request the download link.
  2. Create a soft link:
    ROOT=/path/to/neuralbody
    cd $ROOT/data
    ln -s /path/to/zju_mocap zju_mocap