基于注意力机制及多分支特征融合的实时语义分割算法

• 全文链接
• 请求原文

作   者：

蒋锐;  陈儒娜;  王小明;  李大鹏;  徐友云; 

作者机构：

南京邮电大学通信与信息工程学院; 

关键词：

空间注意力;  特征融合;  自适应注意力;  实时语义分割;  通道注意力; 

期刊名称：

南京邮电大学学报（自然科学版）

i s s n：

年卷期：

2024 年 002 期

页   码：

91-100

摘   要：

为了合理平衡语义分割中的精确度与实时性,基于快速卷积神经网络模型(Fast-SCNN)提出了一种基于注意力机制及多分支特征融合的实时语义分割算法模型.该算法模型首先通过注意力模块捕获空间特征之间的相互联系,增强空间细节信息;然后合理设计融合模块,最大化利用各分支信息,实现深层特征与浅层特征更好的融合;最后引入自适应特征增强注意力模块,捕获长距离像素间的相互依赖关系.实验结果表明,文中算法模型在Cityscapes数据集上获得了71.55%的分割精度,推理速度FPS达到97.6帧/s,模型参数量为1.39 M,验证了该算法所构成网络模型的有效性.