Installation

System/Hardware Requirements

To get started, the following requirements should be fulfilled.

  • System requirements. OpenCOOD is tested under Ubuntu 18.04

  • Adequate GPU. A minimum of 6GB gpu is recommended.

  • Disk Space. Estimate 100GB of space is recommended for data downoading.

  • Python Python3.7 is required.

Installation

1. Dependency Installation

First, download OpenCOOD github to your local folder if you haven’t done it yet.

git clone https://github.com/DerrickXuNu/OpenCOOD.git
cd OpenCOOD

Next we create a conda environment and install the requirements.

conda env create -f environment.yml
conda activate opencood
python setup.py develop

If conda install failed, install through pip

pip install -r requirements.txt

2. Pytorch Installation (>=1.8, tested on 1.8-1.12.0)

Go to https://pytorch.org/ to install pytorch cuda version.

3. Spconv (1.2.1 or 2.x)

OpenCOOD support both spconv 1.2.1 and 2.x to generate voxel features.

To install spconv 1.2.1, please follow the guide in https://github.com/traveller59/spconv/tree/v1.2.1.

To install spconv 2.x, please run the following commands (if you are using cuda 11.3):

pip install spconv-cu113

Tips for installing spconv 1.2.1:

  1. make sure your cmake version >= 3.13.2

  2. CUDNN and CUDA runtime library (use nvcc --version to check) needs to be installed on your machine.

4. Bbx IOU cuda version compile

Install bbx nms calculation cuda version

python opencood/utils/setup.py build_ext --inplace

5. Dependencies for FPV-RCNN (optional)

Install the dependencies for fpv-rcnn.

python opencood/pcdet_utils/setup.py build_ext --inplace