计算机工程

作者：田乐, 王欢

单位：南京理工大学 计算机科学与工程学院, 南京 210094

关键词：语义分割,双模态,注意力机制,特征融合,自适应融合分支

基金：国家自然科学基金（61703209）。

基于可见光、红外双模态数据的场景语义分割在多种复杂环境下较单模态分割显现出更好的性能，然而，获取较好分割效果的前提条件是可见光相机和红外热像仪的成像均清晰。真实场景中存在较多不利的环境因素，如恶劣的光照和天气会对可见光或红外产生不同程度的干扰，从而限制了基于双模态语义分割方法的性能表现。为解决该问题，建立一种改进的双模态语义分割模型。在双流网络架构的基础上增加红外与可见光的像素级融合模块，将其作为一个独立的分支网络并与可见光、红外2个已有分支进行特征级融合，从而实现双模态的像素级和特征级融合。此外，在融合分支中增加空间、通道注意力机制，以挖掘双模态在像素级上的互补特征。实验结果表明，在MF和FR-T这2个公开数据集上，该模型的mIoU指标相比性能表现次优的RTFNet-50模型分别提高6.5和0.6个百分点，且在双模态图像降质和失效时依然具有良好的分割性能。

来源：2022年第8期

《计算机工程》期刊编辑部

查看计算机工程杂志2022年第8期