Laser & Optoelectronics Progress, Volume. 60, Issue 24, 2415003(2023)
Point Cloud Analysis Method Based on Spatial Feature Attention Mechanism
To address the limitations of existing point-based networks, which treat all points with equal emphasis, thereby overlooking crucial features, this paper introduces an attention mechanism to lidar point cloud processing. This mechanism, referred to as the CSA module, integrates the channel attention and spatial attention elements. In a data-driven approach, the two proposed modules autonomously learn the importance of different feature channel information and spatial location information, thereby enhancing the performance of the network on point cloud classification and segmentation tasks. This paper introduces the two modules stated above in a point-based network and proposes a CSA-PointNet++ architecture. The results reveal that the proposed method achieves an accuracy of 93.20% for classification experiments on the ModelNet40 dataset and a mean intersection over union (mIoU) of 82.62% for part segmentation experiments on the ShapeNetPart dataset. This performance is better than that of other comparative methods, indicating the effectiveness of the proposed network. Moreover, classification experiments of the proposed method on a real-world self-constructed dataset yield an accuracy of 92.14%, demonstrating the excellent generalization capability of the proposed network on real-world data.
Get Citation
Copy Citation Text
Yanlin Qu, Yue Wang, Qian Zhang, Shaokun Han. Point Cloud Analysis Method Based on Spatial Feature Attention Mechanism[J]. Laser & Optoelectronics Progress, 2023, 60(24): 2415003
Category: Machine Vision
Received: Mar. 13, 2023
Accepted: May. 15, 2023
Published Online: Nov. 27, 2023
The Author Email: Han Shaokun (skhan@bit.edu.cn)