Spatial-Aware Learning in Feature Embedding and Classification for One-Stage 3-D Object Detection
Yiqiang Wu,Weiping Xiao,Jiantao Gao,Chang Liu,Yu Qin,Yan Peng,Xiaomao Li
DOI: https://doi.org/10.1109/tgrs.2024.3389984
IF: 8.2
2024-04-30
IEEE Transactions on Geoscience and Remote Sensing
Abstract:One-stage 3-D object detection, known for its simplicity and high-speed inference, is attracting increasing attention in autonomous driving scenarios. However, current one-stage detectors tend to perform suboptimally compared to two-stage competitors. Our experimental findings suggest that one-stage detectors underperform due to the underutilization of spatial information in feature embedding and classification. Concretely, the spatial context is severely lost during feature propagation, inducing distorted spatial awareness. On the other hand, category recognition relies on the full utilization of spatial information, which is neglected by current detectors. This inadequate spatial awareness of the classification branch can exacerbate misclassification. To address these issues, we propose spatial-aware learning in feature embedding and classification for one-stage 3-D object detection (SLDet). Specifically, to restore the distorted spatial awareness, categorywise spatial augmentation (CSA) is proposed to adaptively bring the network with pre-encoding multiscale spatial contexts. As for misclassification, spatial guiding classification (SGC) is introduced to guide the classification using explicit scale information. It employs the natural scale divergences among categories to rectify misclassification. Comprehensive experiments demonstrate that SLDet efficiently utilizes spatial information and achieves newly state-of-the-art performance on both the Waymo open dataset and the ONCE dataset. Furthermore, additional experiments demonstrate the excellent generalization capacity of SLDet.
engineering, electrical & electronic,imaging science & photographic technology,remote sensing,geochemistry & geophysics