Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Object DGCNN: 3D Object Detection using Dynamic Graphs

About

3D object detection often involves complicated training and testing pipelines, which require substantial domain knowledge about individual datasets. Inspired by recent non-maximum suppression-free 2D object detection models, we propose a 3D object detection architecture on point clouds. Our method models 3D object detection as message passing on a dynamic graph, generalizing the DGCNN framework to predict a set of objects. In our construction, we remove the necessity of post-processing via object confidence aggregation or non-maximum suppression. To facilitate object detection from sparse point clouds, we also propose a set-to-set distillation approach customized to 3D detection. This approach aligns the outputs of the teacher model and the student model in a permutation-invariant fashion, significantly simplifying knowledge distillation for the 3D detection task. Our method achieves state-of-the-art performance on autonomous driving benchmarks. We also provide abundant analysis of the detection model and distillation framework.

Yue Wang, Justin Solomon• 2021

Related benchmarks

TaskDatasetResultRank
3D Object DetectionnuScenes (val)
NDS67.4
981
3D Object DetectionWaymo Open Dataset (val)
3D APH Vehicle L263.07
219
3D Object DetectionWaymo (val)
Vehicle L2 AP59.27
38
Human Pose EstimationDHP19
MPJPE (2D)6.83
9
Showing 4 of 4 rows

Other info

Code

Follow for update