• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2024, Vol. 46 ›› Issue (05): 872-880.

• Graphics and Images • Previous Articles     Next Articles

Point cloud classification and segmentation based on adaptive graph convolution and attention pooling

LIU Yu-zhen,ZHANG Dong-xia,TAO Zhi-yong   

  1. (School of Electronic and Information Engineering,Liaoning Technology University,Huludao 125105,China)
  • Received:2023-06-07 Revised:2023-10-17 Accepted:2024-05-25 Online:2024-05-25 Published:2024-05-30

Abstract: In response to the limitation of existing point cloud classification and segmentation methods that use max pooling to aggregate local neighborhood features, which leads to the loss of important information beyond the maximum value, this paper proposes a point cloud classification and segmentation network that combines Adaptive Graph Convolution (AGConv) and Attention Pooling (AP). Firstly, a local graph structure of the point cloud is constructed using K-nearest neighbors algorithm, and adaptive convolution kernels are generated based on the features of the points, enabling flexible and accurate capturing of local neighborhood features. Secondly, to effectively enhance feature aggregation, attention pooling is utilized to define an energy function and obtain weight values, which are used to weight and aggregate more representative local features of the point cloud. Finally, adaptive graph convolution and attention pooling are stacked to extract global features layer by layer, thereby improving the accuracy of classification and segmentation. Experimental results demonstrate that compared with the benchmark network, the average class accuracy of point cloud classification is improved by 0.9%, and the average intersection over union of part segmentation and semantic segmentation is improved by 0.8% and 0.3% respectively. This demonstrates that the algorithm can effectively improve the accuracy of point cloud classification and segmentation, and has high robustness.