3D Point Cloud Classification Method Based on Multiple Attention Mechanism and Dynamic Graph Convolution

Authors

  • Yu Zhang
  • Zilong Wang
  • Yongjian Zhu Shanghai Institute of Technology

DOI:

https://doi.org/10.5755/j01.itc.52.3.33035

Keywords:

3D point cloud classification · dynamic graph convolution · self-attention · spatial attention · channel attention

Abstract

In order to solve the problem of uneven density and the low classification accuracy of 3D point cloud, a 3D point cloud classification method fuses multi-attention machine is proposed. It is principally based on the traditional point cloud dynamic graph convolution classification network, into multiple attention mechanisms, including self-attention, spatial attention and channel attention mechanisms. The self-attention mechanism can reduce the dependence on irrelevant points while aligning point clouds, and input the processed point cloud into the classification network. Then the missing geometric information in the classification network is compensated by the integration of spatial and channel attention mechanisms. The experimental results on the public data set ModelNet40 indicate that compared with the DGCNN classification network, the improved network model improves the classification accuracy of the data set by 0.5 % and the average accuracy by 0.9 %. Meantime, the classification accuracy outstrips other contrast classification algorithms.

Downloads

Published

2023-09-26

Issue

Section

Articles