条带池化注意力的实时语义分割算法 |
| |
引用本文: | 吕佳,孙亚南,许鹏程.条带池化注意力的实时语义分割算法[J].计算机辅助设计与图形学学报,2023(9):1395-1404. |
| |
作者姓名: | 吕佳 孙亚南 许鹏程 |
| |
作者单位: | 1. 重庆师范大学计算机与信息科学学院;2. 重庆师范大学重庆市数字农业服务工程技术研究中心 |
| |
基金项目: | 国家自然科学基金重大项目(11991024); |
| |
摘 要: | 针对目前面向注意力机制语义分割算法不能兼顾分割精度与实时性,以及难以满足在现实场景中应用的问题,提出一种基于条带池化注意力的实时语义分割算法.首先采用轻量级骨干网络提取特征信息,结合不同网络层语义差异构建特征融合模块获得不同尺度的上下文信息以提升分割精度;然后利用基于注意力的条带注意力模块(SAM)提高远距离信息被弱化的注意力,并在SAM中加入水平方向的条带池化以降低编码全局上下文的运算量.实验结果表明,所提算法能够得到较高分割精度且满足实时性要求;在Cityscapes测试集上平均交并比为70.6%,分割速度达到了92帧/s;在CamVid测试集上平均交并比为66.4%,分割速度达到了196帧/s.
|
关 键 词: | 现实场景 实时语义分割 注意力机制 条带池化 |
|
|