Installation

June 8, 2021 ยท View on GitHub

git clone https://github.com/limbo0000/InstanceLoc.git

Requirements

  • Linux
  • Python 3.6+
  • PyTorch 1.6+
  • CUDA 10.1+
  • NVCC 2+
  • GCC 4.9+
  • mmcv 0.6.1 (Important!!)

Install MMDetection

The pretraining code is based on MMDet which is further simplified. Here we provide a script install.sh to do it quickly via

sh tools/install.sh

Install Detectron2

The finetuning code is following Detectron2 for the fair comparison. Following this docs to install it or

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