HEDNet: A Hierarchical Encoder-Decoder Network for 3D Object Detection in Point Clouds
About
3D object detection in point clouds is important for autonomous driving systems. A primary challenge in 3D object detection stems from the sparse distribution of points within the 3D scene. Existing high-performance methods typically employ 3D sparse convolutional neural networks with small kernels to extract features. To reduce computational costs, these methods resort to submanifold sparse convolutions, which prevent the information exchange among spatially disconnected features. Some recent approaches have attempted to address this problem by introducing large-kernel convolutions or self-attention mechanisms, but they either achieve limited accuracy improvements or incur excessive computational costs. We propose HEDNet, a hierarchical encoder-decoder network for 3D object detection, which leverages encoder-decoder blocks to capture long-range dependencies among features in the spatial space, particularly for large and distant objects. We conducted extensive experiments on the Waymo Open and nuScenes datasets. HEDNet achieved superior detection accuracy on both datasets than previous state-of-the-art methods with competitive efficiency. The code is available at https://github.com/zhanggang001/HEDNet.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 3D Object Detection | nuScenes (val) | NDS71.4 | 941 | |
| 3D Object Detection | nuScenes (test) | mAP67.7 | 829 | |
| 3D Object Detection | NuScenes v1.0 (test) | mAP67.7 | 210 | |
| 3D Object Detection | nuScenes v1.0 (val) | mAP (Overall)66.7 | 190 | |
| 3D Object Detection | Waymo Open Dataset (val) | 3D APH Vehicle L272.7 | 175 | |
| Semantic segmentation | SemanticKITTI (val) | mIoU70.3 | 117 | |
| 3D Object Detection | Waymo Open Dataset (test) | Vehicle L2 mAPH76.6 | 105 | |
| 3D Object Detection | nuScenes v1.0-trainval (val) | NDS71.4 | 87 | |
| 3D Object Detection | Argoverse 2 (val) | mAP37.1 | 62 | |
| 3D Object Detection | Waymo Open Dataset LEVEL_2 (val) | 3D AP (Overall)75.3 | 46 |