Awesome

Voxel R-CNN

Important Update: The code of Voxel R-CNN in OpenPCDet is also an official implementation one. Please refer to this repository to find the configs for Waymo Open Dataset.

This is the official implementation of Voxel R-CNN: Towards High Performance Voxel-based 3D Object Detection, built on OpenPCDet.

@article{deng2020voxel,
  title={Voxel R-CNN: Towards High Performance Voxel-based 3D Object Detection},
  author={Deng, Jiajun and Shi, Shaoshuai and Li, Peiwei and Zhou, Wengang and Zhang, Yanyong and Li, Houqiang},
  journal={arXiv:2012.15712},
  year={2020}
}

Installation

Prepare for the running environment.

You can either use the docker image we provide, or follow the installation steps in OpenPCDet.
```
docker pull djiajun1206/pcdet-pytorch1.5
```

Prepare for the data.

Please download the official KITTI 3D object detection dataset and organize the downloaded files as follows (the road planes could be downloaded from [road plane], which are optional for data augmentation in the training):

Voxel-R-CNN
├── data
│   ├── kitti
│   │   │── ImageSets
│   │   │── training
│   │   │   ├──calib & velodyne & label_2 & image_2 & (optional: planes)
│   │   │── testing
│   │   │   ├──calib & velodyne & image_2
├── pcdet
├── tools

Generate the data infos by running the following command:

python -m pcdet.datasets.kitti.kitti_dataset create_kitti_infos tools/cfgs/dataset_configs/kitti_dataset.yaml

Setup.
```
python setup.py develop
```

Getting Started

Training.

The configuration file is in tools/cfgs/voxelrcnn, and the training scripts is in tools/scripts.
```
cd tools
sh scripts/train_voxel_rcnn.sh
```
Evaluation.

The configuration file is in tools/cfgs/voxelrcnn, and the training scripts is in tools/scripts.
```
cd tools
sh scripts/eval_voxel_rcnn.sh
```

Acknowledge

Thanks to the strong and flexible OpenPCDet codebase maintained by Shaoshuai Shi (@sshaoshuai) and Chaoxu Guo (@Gus-Guo).

Contact

This repository is implemented by Jiajun Deng (dengjj@mail.ustc.edu.cn).