基于自注意力特征融合组卷积神经网络的三维点云语义分割

摘要

针对现有算法忽略点云数据全局单点特征和局部几何特征的深层关系,导致捕获的局部几何信息缺乏鉴别性且难以有效识别复杂形状的问题,提出基于自注意力特征融合组卷积神经网络的三维点云语义分割算法。首先,设计轻量化网络框架的代理点图卷积提取点云局部几何特征,并加入组卷积操作减少计算量和复杂度,以较少的冗余信息增强特征的丰富性;其次,通过Transformer模块进行不同分支间特征信息的交流,使全局特征和局部几何特征相互补偿,增强特征的完备性;然后,将点云底层语义特征与原始点云融合以扩大局部邻域感受野,获得高级上下文语义信息;最后,将特征输入到分割模块完成细粒度语义分割。实验结果表明,该算法在S3DIS数据集和SemanticKITTI数据集上的分割精度分别达到79.3%和56.6%,能够提取三维点云的关键特征信息,网络参数量较少且具有较高的语义分割鲁棒性。