Supervoxel Convolution for Online 3D Semantic Segmentation

• Published in: ACM transactions on graphics 2021-06, Vol.40 (3), p.1-15, Article 34
• Main Authors: Huang, Shi-Sheng, Ma, Ze-Yu, Mu, Tai-Jiang, Fu, Hongbo, Hu, Shi-Min
• Format: Article
• Language: English
• Subjects: Artificial intelligence; Computer Graphics; Computing methodologies; Machine Learning
• ISSN: 0730-0301; 1557-7368
• DOI: 10.1145/3453485

Online 3D semantic segmentation, which aims to perform real-time 3D scene reconstruction along with semantic segmentation, is an important but challenging topic. A key challenge is to strike a balance between efficiency and segmentation accuracy. There are very few deep-learning-based solutions to this problem, since the commonly used deep representations based on volumetric-grids or points do not provide efficient 3D representation and organization structure for online segmentation. Observing that on-surface supervoxels, i.e., clusters of on-surface voxels, provide a compact representation of 3D surfaces and brings efficient connectivity structure via supervoxel clustering, we explore a supervoxel-based deep learning solution for this task. To this end, we contribute a novel convolution operation (SVConv) directly on supervoxels. SVConv can efficiently fuse the multi-view 2D features and 3D features projected on supervoxels during the online 3D reconstruction, and leads to an effective supervoxel-based convolutional neural network, termed as Supervoxel-CNN, enabling 2D-3D joint learning for 3D semantic prediction. With the Supervoxel-CNN, we propose a clustering-then-prediction online 3D semantic segmentation approach. The extensive evaluations on the public 3D indoor scene datasets show that our approach significantly outperforms the existing online semantic segmentation systems in terms of efficiency or accuracy.