图形图像技术 | 融合通道门控与特征交互的点云语义分割网络
图形图像技术 | 融合通道门控与特征交互的点云语义分割网络
摘 要:针对基于深度学习的三维点云处理方法在语义分割任务中缺失局部特征的问题,提出了融合通道门控与特征交互的点云语义分割网络(point cloud semantic segmentation network based on channel gating and feature interaction, CGINet)。该网络引入特征交互模块(feature interaction module, FIM), 设计轻量通道门控模块(light channel gate, LCG), 用于对特征通道进行自适应调节, 以增强局部响应能力, 建模点之间的上下文关系, 从而提升点云语义表示能力。同时构建全局通道加权模块(global channel weighting, GCW), 对通道级语义信息进行压缩建模与加权融合, 进一步强化全局特征感知能力。此外, 网络中引入残差结构, 有效提升了特征传递的稳定性与表达能力。在 S3DIS(Stanford large-scale 3D indoor spaces) 和 ShapeNetPart 数据集上分别对所提模型进行了实验。在 S3DIS 数据集的语义分割实验中平均交并比(mIoU)相较于 PointNet++ 提高了 9.7 个百分点, 在 ShapeNetPart 数据集的语义分割实验中相较于 PointNet++ 类别平均交并比(class average mIoU) 和实例平均交并比 (instance average mIoU) 分别提升了 2.3 和 2.2 个百分点。实验结果表明, 该方法在两组实验中均实现了更优的分割效果, 验证了所提网络结构的有效性。