arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2108.06925cs.CV

Interpolation-Aware Padding for 3D Sparse Convolutional Neural Networks

Tsinghua University(清华大学);Microsoft Research Asia(微软亚洲研究院)

Yu-Qi Yang, Peng-Shuai Wang, Yang Liu

英文摘要:

Sparse voxel-based 3D convolutional neural networks (CNNs) are widely used for various 3D vision tasks. Sparse voxel-based 3D CNNs create sparse non-empty voxels from the 3D input and perform 3D convolution operations on them only. We propose a simple yet effective padding scheme --- interpolation-aware padding to pad a few empty voxels adjacent to the non-empty voxels and involve them in the 3D CNN computation so that all neighboring voxels exist when computing point-wise features via the trilinear interpolation. For fine-grained 3D vision tasks where point-wise features are essential, like semantic segmentation and 3D detection, our network achieves higher prediction accuracy than the existing networks using the nearest neighbor interpolation or the normalized trilinear interpolation with the zero-padding or the octree-padding scheme. Through extensive comparisons on various 3D segmentation and detection tasks, we demonstrate the superiority of 3D sparse CNNs with our padding scheme in conjunction with feature interpolation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑