PMP-Net plus plus : Point Cloud Completion by Transformer-Enhanced Multi-Step Point Moving Paths

被引:106
作者
Wen, Xin [1 ,2 ]
Xiang, Peng [1 ]
Han, Zhizhong [3 ]
Cao, Yan-Pei [4 ]
Wan, Pengfei
Zheng, Wen
Liu, Yu-Shen [5 ]
机构
[1] Tsinghua Univ, Sch Software, Beijing 100190, Peoples R China
[2] JDcom, JD Logist, Beijing 101111, Peoples R China
[3] Wayne State Univ, Dept Comp Sci, Detroit, MI 48202 USA
[4] Kuaishou Technol, Y tech, Beijing 100085, Peoples R China
[5] Tsinghua Univ, Sch Software, BNRist, Beijing 100190, Peoples R China
基金
国家重点研发计划; 中国国家自然科学基金;
关键词
Point clouds; 3D shape completion; transformer; up-sampling;
D O I
10.1109/TPAMI.2022.3159003
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Point cloud completion concerns to predict missing part for incomplete 3D shapes. A common strategy is to generate complete shape according to incomplete input. However, unordered nature of point clouds will degrade generation of high-quality 3D shapes, as detailed topology and structure of unordered points are hard to be captured during the generative process using an extracted latent code. We address this problem by formulating completion as point cloud deformation process. Specifically, we design a novel neural network, named PMP-Net++, to mimic behavior of an earth mover. It moves each point of incomplete input to obtain a complete point cloud, where total distance of point moving paths (PMPs) should be the shortest. Therefore, PMP-Net++ predicts unique PMP for each point according to constraint of point moving distances. The network learns a strict and unique correspondence on point-level, and thus improves quality of predicted complete shape. Moreover, since moving points heavily relies on per-point features learned by network, we further introduce a transformer-enhanced representation learning network, which significantly improves completion performance of PMP-Net++. We conduct comprehensive experiments in shape completion, and further explore application on point cloud up-sampling, which demonstrate non-trivial improvement of PMP-Net++ over state-of-the-art point cloud completion/up-sampling methods.
引用
收藏
页码:852 / 867
页数:16
相关论文
共 57 条
[1]  
[Anonymous], 2019, P 27 ACM INT C MULT, DOI DOI 10.1145/3343031.3350960
[2]  
Berger M., 2014, Eurographics 2014state of the art reports, V1, P161, DOI DOI 10.2312/EGST.20141040
[3]   Structure Recovery by Part Assembly [J].
Shen, Chao-Hui ;
Fu, Hongbo ;
Chen, Kang ;
Hu, Shi-Min .
ACM TRANSACTIONS ON GRAPHICS, 2012, 31 (06)
[4]   Shape Completion using 3D-Encoder-Predictor CNNs and Shape Synthesis [J].
Dai, Angela ;
Qi, Charles Ruizhongtai ;
Niessner, Matthias .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :6545-6554
[5]   A Field Model for Repairing 3D Shapes [J].
Duc Thanh Nguyen ;
Hua, Binh-Son ;
Minh-Khoi Tran ;
Quang-Hieu Pham ;
Yeung, Sai-Kit .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :5676-5684
[6]   A Papier-Mache Approach to Learning 3D Surface Generation [J].
Groueix, Thibault ;
Fisher, Matthew ;
Kim, Vladimir G. ;
Russell, Bryan C. ;
Aubry, Mathieu .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :216-224
[7]   High-Resolution Shape Completion Using Deep Neural Networks for Global Structure and Local Geometry Inference [J].
Han, Xiaoguang ;
Li, Zhen ;
Huang, Haibin ;
Kalogerakis, Evangelos ;
Yu, Yizhou .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :85-93
[8]  
Han ZZ, 2020, PR MACH LEARN RES, V119
[9]  
Han ZZ, 2019, PROCEEDINGS OF THE TWENTY-EIGHTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P758
[10]  
Han ZZ, 2020, Img Proc Comp Vis Re, V12369, P607, DOI 10.1007/978-3-030-58586-0_36