End-to-End Point Cloud Completion Network with Attention Mechanism

Sensors (Basel). 2022 Aug 26;22(17):6439. doi: 10.3390/s22176439.

Abstract

We propose a conceptually simple, general framework and end-to-end approach to point cloud completion, entitled PCA-Net. This approach differs from the existing methods in that it does not require a "simple" network, such as multilayer perceptrons (MLPs), to generate a coarse point cloud and then a "complex" network, such as auto-encoders or transformers, to enhance local details. It can directly learn the mapping between missing and complete points, ensuring that the structure of the input missing point cloud remains unchanged while accurately predicting the complete points. This approach follows the minimalist design of U-Net. In the encoder, we encode the point clouds into point cloud blocks by iterative farthest point sampling (IFPS) and k-nearest neighbors and then extract the depth interaction features between the missing point cloud blocks by the attention mechanism. In the decoder, we introduce a new trilinear interpolation method to recover point cloud details, with the help of the coordinate space and feature space of low-resolution point clouds, and missing point cloud information. This paper also proposes a method to generate multi-view missing point cloud data using a 3D point cloud hidden point removal algorithm, so that each 3D point cloud model generates a missing point cloud through eight uniformly distributed camera poses. Experiments validate the effectiveness and superiority of PCA-Net in several challenging point cloud completion tasks, and PCA-Net also shows great versatility and robustness in real-world missing point cloud completion.

Keywords: deep learning; end to end; point cloud completion; squeeze and excitation; trilinear interpolation.

MeSH terms

  • Algorithms*
  • Cluster Analysis
  • Electric Power Supplies
  • Neural Networks, Computer*
  • Research Design