基于注意力机制的全卷积神经网络模型

• 作者: 刘孟轩 张蕊 曾志远 金玮 武益超
• 来源期刊: 现代信息科技
• 学科: 工学
• 关键词: 语义分割 全卷积神经网络 注意力机制 特征融合
• 年，卷（期）: 2021,（23）
• 所属期刊栏目: 计算机技术|Computer Technology
• 页码范围: 92-95
• 页数: 4页
• 分类号: TP391.4
• 语种: 中文
• DOI: 10.19850/j.cnki.2096-4706.2021.23.024

摘要

全卷积神经网络FCN-8S在进行多尺度特征融合时,由于未能考虑不同尺度特征各自的特点进行充分融合,导致分割结果精度较低,针对这一问题,文章提出了一种基于注意力机制的多尺度特征融合的全卷积神经网络模型.该模型基于注意力机制对FCN-8S中的不同尺度特征进行加权特征融合,以相互补充不同尺度特征包含的不同信息,进而提升网络的分割效果.文章模型在公共数据集PASCAL VOC2012和Cityscapes上进行验证,MIoU相对于FCN-8S分别提升了2.2％和0.8％.