PIG-Net: Inception base deep learning architecture for 3D point cloud segmentation
Sindhu Hegde Shankar Gangisetty
IIIT Hyderabad [Code] [Paper] [arXiv] KLE Technological University
Computers & Graphics 2021
PIG-Net: Point Inception Global average pooling Network
Abstract
Point clouds, being the simple and compact representation of surface geometry of 3D objects, have gained increasing popularity with the evolution of deep learning networks for classification and segmentation tasks. Unlike human, teaching the machine to analyze the segments of an object is a challenging task and quite essential in various machine vision applications. In this paper, we address the problem of segmentation and labelling of the 3D point clouds by proposing a inception based deep network architecture called PIG-Net, that effectively characterizes the local and global geometric details of the point clouds. In PIG-Net, the local features are extracted from the transformed input points using the proposed inception layers and then aligned by feature transform. These local features are aggregated using the global average pooling layer to obtain the global features. Finally, feed the concatenated local and global features to the convolution layers for segmenting the 3D point clouds. We perform an exhaustive experimental analysis of the PIG-Net architecture on two state-of-the-art datasets, namely, ShapeNet [1] and PartNet [2]. We evaluate the effectiveness of our network by performing ablation study.
Results
Segmentation results of ShapeNet-part
PartNet (coarse level) dataset
Segmentation results of PartNet
Visualization of 3D Part Segmentation Results
Citation
If you find PIG-Net useful, please consider citing:
@article{HEGDE202113,
author = "Sindhu Hegde and Shankar Gangisetty",
title = "PIG-Net: Inception based deep learning architecture for 3D point cloud segmentation",
journal = "Computers & Graphics",
volume = "95",
pages = "13 - 22",
year = "2021",
issn = "0097-8493",
doi = "https://doi.org/10.1016/j.cag.2021.01.004"
}
Contact
Sindhu Hegde - sindhu.hegde@research.iiit.ac.in
Shankar Gangisetty - shankar@kletech.ac.in