1
|
Wu Y, Chen X, Huang X, Song K, Zhang D. Unsupervised distribution-aware keypoints generation from 3D point clouds. Neural Netw 2024; 173:106158. [PMID: 38340470 DOI: 10.1016/j.neunet.2024.106158] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [What about the content of this article? (0)] [Affiliation(s)] [Abstract] [Key Words] [MESH Headings] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 08/21/2023] [Revised: 12/07/2023] [Accepted: 02/01/2024] [Indexed: 02/12/2024]
Abstract
Keypoints extraction from 3D objects is a fundamental task in point cloud processing. The ideal keypoints should be an ordered and well-aligned set of points that effectively reflect the shape and structure of the object. To this end, this paper proposes an unsupervised 3D point cloud keypoints generation network with the consideration of the probability distribution of keypoints and spatial distribution among keypoints. The network downsamples and groups the 3D point cloud, obtaining local features of the point cloud. The local features are leveraged to explicitly learn the mixture probability distribution of keypoint position. A composite loss function that comprehensively considers shape similarity, point importance, and geometric constraint is proposed to guide the network in generating keypoints with semantic consistency and regular spatial distribution. The experimental results and quantitative comparisons on the ShapeNet and KeypointNet datasets demonstrate that the proposed method achieves ordered, well-aligned, and robust keypoints generation for 3D point clouds. The source code of the proposed method is available at https://github.com/djzgroup/Keypoints.
Collapse
Affiliation(s)
- Yiqi Wu
- School of Computer Science, China University of Geoscience, NO.68 Jincheng Street, Wuhan, 430078, Hubei, China; Hubei Key Laboratory of Intelligent Geo-Information Processing, China University of Geosciences, NO.68 Jincheng Street, Wuhan, 430078, Hubei, China
| | - Xingye Chen
- School of Electronic Information and Communications, Huazhong University of Science and Technology, NO.1037 Luoyu Road, Wuhan, 430074, Hubei, China
| | - Xuan Huang
- School of Computer Science, China University of Geoscience, NO.68 Jincheng Street, Wuhan, 430078, Hubei, China
| | - Kelin Song
- School of Computer Science, China University of Geoscience, NO.68 Jincheng Street, Wuhan, 430078, Hubei, China
| | - Dejun Zhang
- School of Computer Science, China University of Geoscience, NO.68 Jincheng Street, Wuhan, 430078, Hubei, China; Hubei Key Laboratory of Intelligent Geo-Information Processing, China University of Geosciences, NO.68 Jincheng Street, Wuhan, 430078, Hubei, China; Information Systems Technology and Design Pillar, Singapore University of Technology and Design, 8 Somapah Road, 487372, Singapore.
| |
Collapse
|