/PDV

Point Density-Aware Voxels for LiDAR 3D Object Detection (CVPR 2022)

Primary LanguagePythonApache License 2.0Apache-2.0

PDV

PDV is LiDAR 3D object detection method. This repository is based off [OpenPCDet].

Point Density-Aware Voxels for LiDAR 3D Object Detection
Jordan S. K. Hu, Tianshu Kuai, Steven L. Waslander
[Paper]

PDV diagram

Overview

Changelog

[2022-03-07] PDV v0.1.0 is released.

Model Zoo

KITTI 3D Object Detection Baselines

Selected supported methods are shown in the below table. The results are the 3D detection performance of moderate difficulty on the val set of KITTI dataset.

  • All models are trained with 2 NVIDIA Tesla P100 GPUs and are available for download.
  • The training time is measured with 2 NVIDIA Tesla P100 GPUs and PyTorch 1.7.
training time Car@R40 Pedestrian@R40 Cyclist@R40 log download
PDV ~23 hours 85.05 57.41 75.95 log model-147M

Waymo Open Dataset Baselines

We could not provide the above pretrained models due to Waymo Dataset License Agreement, but you could easily achieve similar performance by training with the default configs. PDV is trained with 10% data (~16k frames) on 4 NVIDIA Tesla V100s GPUs.

Installation

Please refer to INSTALL.md for the installation of PDV.

Quick Demo

Please refer to DEMO.md for a quick demo to test with a pretrained model and visualize the predicted results on your custom data or the original KITTI data.

Getting Started

Please refer to GETTING_STARTED.md to learn more usage about this project.

License

PDV is released under the Apache 2.0 license.

Acknowledgement

We would like to thank the authors of OpenPCDet for their open source release of their codebase.

Citation

If you find this project useful in your research, please consider citing:

@article{PDV,
    title={Point Density-Aware Voxels for LiDAR 3D Object Detection},
    author={Jordan S. K. Hu and
            Tianshu Kuai and
            Steven L. Waslander},
    journal={CVPR},
    year={2022}
}