shellnet

Deep learning with 3D data has progressed significantly since the introduction of convolutional neural networks that can handle point order ambiguity in point cloud data. While being able to achieve good accuracies in various scene understanding tasks, previous methods often have low training speed and complex network architecture. In this paper, we address these problems by proposing an efficient end-to-end permutation invariant convolution for point cloud deep learning. Our simple yet effective convolution operator named ShellConv uses statistics from concentric spherical shells to define representative features and resolve the point order ambiguity, allowing traditional convolution to perform on such features. Based on ShellConv we further build an efficient neural network named ShellNet to directly consume the point clouds with larger receptive fields while maintaining less layers. We demonstrate the efficacy of ShellNet by producing state-of-the-art results on object classification, object part segmentation, and semantic scene segmentation while keeping the network very fast to train.

Materials


Paper
   
Video
   
Poster
   
Code

Citation

If you find this work useful, please cite the following paper.

@inproceedings{zhang2019shellnet,
  title={Shellnet: Efficient Point Cloud Convolutional Neural Networks Using Concentric Shells Statistics},
  author={Zhang, Zhiyuan and Hua, Binh-Son and Yeung, Sai-Kit},
  booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
  pages={1607--1616},
  year={2019}
}

Acknowledgements

The authors acknowledge support from the SUTD Digital Manufacturing and Design Centre funded by the Singapore National Research Foundation, and an internal grant from HKUST (R9429).