GT-Net: Variational Autoencoder Networks based on Graph Transformer for 3D Shape Learning

Zhenjiang Du,Yi Lu,Guan Wang,Ning Xie,Yang Yang
DOI: https://doi.org/10.1109/ICME55011.2023.00162
2023-01-01
Abstract:In this paper, we introduce a novel structure-aware method to generate diverse, and realistic 3D shapes via semantic parts. Most previous works neglect the structural and context information between shape parts and only consider the geometric information. This sometimes leads to the wrong combination of parts during the generation process and brought down the generation quality. To address this issue, we learn a structure-aware latent representation for 3D shapes by training a variational autoencoder(VAE). Specially, we use a graph to express semantic parts and their structural relationship of the 3D shape. Based on that graph representation, we design a generative network based on graph transformer architecture, called Graph Transformer VAE networks(GT-Net), to encode and decode the graph-represented 3D shape. Our experimental results demonstrate that our method achieves better performance than previous methods among various shape families, especially in terms of capturing shape details information.
What problem does this paper attempt to address?