Context-based local-global fusion network for 3D point cloud classification and segmentation

被引:4
作者
Wu, Junwei [2 ,3 ]
Sun, Mingjie [1 ]
Jiang, Chenru [3 ]
Liu, Jiejie [3 ]
Smith, Jeremy [2 ]
Zhang, Quan [3 ]
机构
[1] Soochow Univ, Dept Comp Sci & Technol, Suzhou, Peoples R China
[2] Univ Liverpool, Dept Elect Engn & Elect, Liverpool L69 3GY, England
[3] Xian Jiaotong Liverpool Univ, Dept Elect & Elect Engn, Suzhou 215123, Peoples R China
关键词
Point cloud; Context learning; Global attention; Local-global fusion;
D O I
10.1016/j.eswa.2024.124023
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
3D point clouds have gained much research attention because of their ability to represent the spatial information of real-world environments in a detailed manner. Despite recent progress in point cloud processing with deep neural networks, most of them either implement sophisticated local feature aggregation methods or imitate 2D convolution operations in the range of K nearest neighbors with limited local context information. These methods may struggle to distinguish between similar geometric shapes within the local region of K nearest neighbors, such as doors and walls. To address this issue, we propose a novel local-global fusion network that captures the diverse local geometric shapes with global structure information. The proposed local-global fusion network comprises two main modules. Firstly, we have developed an effective approach for local context learning using incremental dilated KNN (IDKNN) as the neighbor selecting mechanism to enlarge the receptive field and incorporate more reliable points for local geometric shape learning. Secondly, a three-direction region-wise spatial attention (TRSA) algorithm has been developed to explore the global contextual dependencies. For global context learning, we first split the entire 3D space into regions with equal numbers of points, and, then, intra-region context features are extracted to learn the inter-region relations from three orthogonal directions, taking global structural knowledge into account. By fusing the local context information and global contextual dependencies, we establish a Local-Global Fusion Network, end-to-end framework, called LGFNet. Extensive experimental results on several benchmark datasets clearly demonstrate our approach can achieve state-of-the-art (SOTA) performance on point cloud classification, part segmentation, and indoor semantic segmentation. In addition, TRSA and IKDNN can be easily used in a plug-and-play fashion with various existing SOTA networks to substantially improve their performance. Our code is available at https://github.com/jasonwjw/IDKNN
引用
收藏
页数:12
相关论文
共 53 条
  • [51] TMFNet: Three-Input Multilevel Fusion Network for Detecting Salient Objects in RGB-D Images
    Zhou, Wujie
    Pan, Sijia
    Lei, Jingsheng
    Yu, Lu
    [J]. IEEE TRANSACTIONS ON EMERGING TOPICS IN COMPUTATIONAL INTELLIGENCE, 2022, 6 (03): : 593 - 601
  • [52] Salient Object Detection in Stereoscopic 3D Images Using a Deep Convolutional Residual Autoencoder
    Zhou, Wujie
    Wu, Junwei
    Lei, Jingsheng
    Hwang, Jenq-Neng
    Yu, Lu
    [J]. IEEE TRANSACTIONS ON MULTIMEDIA, 2021, 23 (23) : 3388 - 3399
  • [53] Cylindrical and Asymmetrical 3D Convolution Networks for LiDAR Segmentation
    Zhu, Xinge
    Zhou, Hui
    Wang, Tai
    Hong, Fangzhou
    Ma, Yuexin
    Li, Wei
    Li, Hongsheng
    Lin, Dahua
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 9934 - 9943