TMSDNet: Transformer with multi‐scale dense network for single and multi‐view 3D reconstruction

• Published in: Computer animation and virtual worlds Vol. 35; no. 1
• Main Authors: Zhu, Xiaoqiang, Yao, Xinsheng, Zhang, Junjie, Zhu, Mengyao, You, Lihua, Yang, Xiaosong, Zhang, Jianjun, Zhao, He, Zeng, Dan
• Format: Journal Article
• Language: English
• Published: Chichester Wiley Subscription Services, Inc 01.01.2024
• Subjects: deep learning; Feature extraction; Image reconstruction; multi‐scale; single‐view and multi‐view 3D reconstruction; transformer; Transformers
• DOI: 10.1002/cav.2201
• ISSN: 1546-4261

3D reconstruction is a long‐standing problem. Recently, a number of studies have emerged that utilize transformers for 3D reconstruction, and these approaches have demonstrated strong performance. However, transformer‐based 3D reconstruction methods tend to establish the transformation relationship between the 2D image and the 3D voxel space directly using transformers or rely solely on the powerful feature extraction capabilities of transformers. They ignore the crucial role played by deep multi‐scale representation of the object in the voxel feature domain, which can provide extensive global shape and local detail information about the object in a multi‐scale manner. In this article, we propose a novel framework TMSDNet (transformer with multi‐scale dense network) for single‐view and multi‐view 3D reconstruction with transformer to solve this problem. Based on our well‐designed combined‐transformer Block, which is canonical encoder–decoder architecture, voxel features with spatial order can be extracted from the input image, which are used to further extract multi‐scale global features in parallel using a multi‐scale residual attention module. Furthermore, a residual dense attention block is introduced for deep local features extraction and adaptive fusion. Finally, the reconstructed objects are produced with the voxel reconstruction block. Experiment results on the benchmarks such as ShapeNet and Pix3D datasets demonstrate that TMSDNet outperforms the existing state‐of‐the‐art reconstruction methods substantially. This paper proposes a novel 3D reconstruction network TMSDNet, which uses transformers' capability of strong feature extraction and processing of relative order between features to obtain voxel features. With the multiple bypass and RDAB in MSRAM, TMSDNet can utilize the information about the global shape and local details in the deep multi‐scale representation of the object in the voxel feature domain to further improve the performance. Extensive experiments show TMSDNet has better reconstruction performance, fewer parameters and competitive inference time.