Probabilistic Boundary-Guided Point Cloud Primitive Segmentation Network
Shaohu Wang, Fangbo Qin, Yuchuang Tong, Xiuqin Shang, Zhengtao Zhang
Abstract
3D point cloud primitive segmentation aims to segment an original entire point cloud into a set of geometric primitives with different types, which is widely used in the manufacturing industry. The primitive segmentation task is challenging faced with complex shapes and ambiguous boundaries. We observe that the boundary properties of point clouds have not been fully investigated and exploited in previous works, and the primitive segmentation performance is not satisfactory near boundaries, especially gradually-changed boundaries. In this paper, we propose a novel probabilistic boundary guided primitive segmentation network (PBPS) to improve the primitive segmentation ability by emphasizing the boundary cues. Firstly, the point cloud boundary is represented by Gaussian distribution instead of binary representation, which can describe boundaries more informatively and also provides an indication of the ambiguous relationship between point cloud boundaries and inner regions. Second, a probabilistic boundary-guided feature fusion module as well as an instance clustering and type voting strategy are proposed, which process the boundary points and non-boundary points conditioning on different boundary probabilities, to reduce the impact of boundary ambiguity on primitive segmentation. Thirdly, a primitive instance contrastive loss is designed which can relatively loosen the constraints on the distances from boundary points to centroid in the embedding space. The effectiveness of PBPS was verified by a series of experiments on two CAD-based datasets and two real-scene dataset.