Skip to content

Latest commit

 

History

History
32 lines (27 loc) · 1003 Bytes

INSTALL.md

File metadata and controls

32 lines (27 loc) · 1003 Bytes

Requirements

Our code is tested on

  • Python 3.10
  • Pytorch 1.11
  • Detectron 0.6

Installation guide:

  1. Install PyTorch 1.11
  2. install the cython package (e.g. pip install cython)
  3. Clone the official detectron2 repository, e.g. at commit 45b3fce:
    • git clone detectron2
    • git checkout 45b3fce
  4. Install detectron2, e.g. via python setup.py develop while inside detectron2 repository
  5. Inside DSG directory, run python setup.py develop
  6. Install modified pycocotools:
    • go to subdirectory cocoapi/PythonAPI
    • python setup.py build_ext install
  7. Install additional required packages:
  • opencv-python
  • scikit-learn
  • networkx
  • imantics
  • easydict
  • h5py
  • numpy < 1.24
  • networkx == 2.8
  • shapely

Note: Please make sure the detectron2 version correctly corresponds to the pytorch and cuda versions installed. Please check this page for installation instructions and common issues.