Skip to content

Latest commit

 

History

History
49 lines (41 loc) · 1.25 KB

INSTALL.md

File metadata and controls

49 lines (41 loc) · 1.25 KB
Raw

Installation

Requirements

  • Python >= 3.6
  • PyTorch >= 1.5
  • GCC >= 4.9
  • cython: pip install -U cython
  • simplejson: pip install simplejson
  • PyAV: conda install av -c conda-forge
  • iopath: pip install -U iopath or conda install -c iopath iopath
  • psutil: pip install psutil
  • OpenCV: pip install opencv-python
  • tensorboard: pip install tensorboard
  • moviepy: (optional, for visualizing video on tensorboard) conda install -c conda-forge moviepy or pip install moviepy

Install Detectron2

git clone https://github.com/facebookresearch/detectron2.git
python -m pip install -e detectron2

If have to reinstall, first:

rm -rf build/ **/*.so

Install Apex (optional)

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

For old env, (e.g. cuda9.0, torch.15), you should rollback apex to earlier version

# https://github.com/NVIDIA/apex/issues/802#issuecomment-618699214
git checkout f3a960f80244cf9e80558ab30f7f7e8cbf03c0a0

Build

After having the above dependencies, run:

git clone https://github.com/ShoufaChen/WOO.git
cd WOO
python setup.py build develop

Now the installation is finished.