Three-dimensional object detection network based on geometric information supplement strategy

doi:10.1016/j.compeleceng.2023.108577

Three-dimensional object detection network based on geometric information supplement strategy

2023-03-01

3

PRE

AI

DOI：10.1016/j.compeleceng.2023.108577

原文链接

原文求助

摘要

En 中文

Three-dimensional object detection from point clouds greatly advances autonomous driving. The real point clouds are unbalanced, where some occluded or distant objects with fewer points suffer from insufficient features, resulting in limited detection accuracy. To address this issue, we propose a novel three-dimensional object detection network based on a Geometric Information Supplement (GIS) strategy. Concretely, we design a novel shape completion model and a point attention fusion module to implement the GIS strategy. The shape completion model based on a structure-aware Transformer first recovers the complete shape points of objects within proposals. Then, the point attention fusion module supplies the recovered geometric shape cues to the original spatial features of objects. By the GIS strategy, the network learns enriched geometric features of objects for proposal refinement, thus, generating accurate detection boxes. Comparative experiments on KITTI dataset show that our method significantly outperforms other state-ofthe-art methods while maintaining modest model complexity.

Keyword:

Three-dimensional object detection

Shape completion

Transformer

Attention mechanism

Feature fusion

Three-dimensional object detection network based on geometric information supplement strategy

摘要

期刊

机构

引用论文