Skip to content

Latest commit

 

History

History
34 lines (27 loc) · 1.21 KB

INSTALL.md

File metadata and controls

34 lines (27 loc) · 1.21 KB

Installation

Modified from Det3D.

Requirements

  • Python 3.6 or higher
  • PyTorch 1.1 or higher
  • CUDA 10.0 or higher
  • CMake 3.13.2 or higher

Setup

git clone https://github.com/qcraftai/simtrack.git
cd simtrack
pip install -r requirements.txt
  • Compile CUDA code for IOU3D_NMS.
cd det3d/ops/iou3d_nms
python setup.py build_ext --inplace
  • If you want to use SyncBN, please install APEX.
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 the voxel based backbone, please install SpConv.

Note: Our voxel based model was trained and tested based on SpConv1.x and may not work with the latest version. We are working on this to solve the conflict, please stay tuned.