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

被引:112
作者
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 条
[41]   SoftPoolNet: Shape Descriptor for Point Cloud Completion and Classification [J].
Wang, Yida ;
Tan, David Joseph ;
Navab, Nassir ;
Tombari, Federico .
COMPUTER VISION - ECCV 2020, PT III, 2020, 12348 :70-85
[42]   Patch-based Progressive 3D Point Set Upsampling [J].
Wang Yifan ;
Shihao Wu ;
Hui Huang ;
Cohen-Or, Daniel ;
Sorkine-Hornung, Olga .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :5951-5960
[43]   CF-SIS: Semantic-Instance Segmentation of 3D Point Clouds by Context Fusion with Self-Attention [J].
Wen, Xin ;
Han, Zhizhong ;
Youk, Geunhyuk ;
Liu, Yu-Shen .
MM '20: PROCEEDINGS OF THE 28TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, 2020, :1661-1669
[44]   PMP-Net: Point Cloud Completion by Learning Multi-step Point Moving Paths [J].
Wen, Xin ;
Xiang, Peng ;
Han, Zhizhong ;
Cao, Yan-Pei ;
Wan, Pengfei ;
Zheng, Wen ;
Liu, Yu-Shen .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :7439-7448
[45]   Cycle4Completion: Unpaired Point Cloud Completion using Cycle Transformation with Missing Region Coding [J].
Wen, Xin ;
Han, Zhizhong ;
Cao, Yan-Pei ;
Wan, Pengfei ;
Zheng, Wen ;
Liu, Yu-Shen .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :13075-13084
[46]   Point Cloud Completion by Skip-attention Network with Hierarchical Folding [J].
Wen, Xin ;
Li, Tianyang ;
Han, Zhizhong ;
Liu, Yu-Shen .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1936-1945
[47]   Point2SpatialCapsule: Aggregating Features and Spatial Relationships of Local Regions on Point Clouds Using Spatial-Aware Capsules [J].
Wen, Xin ;
Han, Zhizhong ;
Liu, Xinhai ;
Liu, Yu-Shen .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 :8855-8869
[48]   GRNet: Gridding Residual Network for Dense Point Cloud Completion [J].
Xie, Haozhe ;
Yao, Hongxun ;
Zhou, Shangchen ;
Mao, Jiageng ;
Zhang, Shengping ;
Sun, Wenxiu .
COMPUTER VISION - ECCV 2020, PT IX, 2020, 12354 :365-381
[49]   FoldingNet: Point Cloud Auto-encoder via Deep Grid Deformation [J].
Yang, Yaoqing ;
Feng, Chen ;
Shen, Yiru ;
Tian, Dong .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :206-215
[50]   Semantic Image Inpainting with Deep Generative Models [J].
Yeh, Raymond A. ;
Chen, Chen ;
Lim, Teck Yian ;
Schwing, Alexander G. ;
Hasegawa-Johnson, Mark ;
Do, Minh N. .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :6882-6890