HGG-Net: Hierarchical Geometry Generation Network for Point Cloud Completion

Point cloud completion concerns the inference of the completed geometries for real-scanned point clouds that are sparse and incomplete due to occlusion, noise, and viewpoint. Previous methods usually learn a one-shot partial-to-complete mapping, which is incapable of generating fine structure details for the complex point cloud distributions. In this paper, a Hierarchical Geometry Generation point completion Network (HGG-Net) is proposed to hierarchically generate the fine-grained completed point cloud with a skeleton-to-details strategy, which consists of three fundamental modules, namely Transformer-enhanced Feature Encoder (TFE), Multi-level Geometry Representation Decoder (MGRD), and Hierarchical Dynamic Geometry Generator (HDG). Specifically, TFE first extracts geometry features of the incomplete input and obtains a coarse prediction via self-attention mechanism and edge convolution. Second, MGRD obtains the multi-level decoded geometry representations by Geometric Interactive Transformer (GIT) and Channel-Attention-based Geometry Features Fusion (CAGF), where GIT is proposed to decode the complete prompt by capturing the semantic relationship between geometry features of the incomplete and the decoded complete objects, and CAGF aims to fuse them for the high-quality representation. Third, HDG generates the complete points hierarchically from skeleton to details based on the Dynamic Graph Attention mechanism. Qualitative and quantitative experiments demonstrate that the proposed HGG-Net outperforms state-of-the-art methods on several point cloud completion datasets. Our code is available at https://github.com/haalexx/HGGNet.