快速超轻量城市交通场景语义分割

摘要：

近年来,随着自动驾驶的火热发展,越来越多研究者开始探索图像语义分割网络的轻量化并将其应用于道路交通场景.而目前现存的语义分割网络通常由于参数量庞大难以部署在硬件资源有限的边缘设备,针对这一问题,设计了一个由通道注意力骨干网络(CABb)和空间注意力解码器(SAD)模块构成的双注意力轻量化网络(DALNet),结合"通道-空间"双注意力机制的DALNet在图像上下文语义信息的提取和图像空间信息的恢复上都具有突出的表现.CABb主要由通道注意力瓶颈(CABt)模块组成,CABt模块采用Split策略分离特征通道并行处理多尺度的特征图,引入通道注意力机制进行通道融合,提取多尺度语义信息.SAD模块利用空间注意力机制指导解码器进行双线性插值上采样,恢复分割目标边沿以及细节信息.实验结果表明,DALNet仅凭48万的参数量在城市交通数据集Cityscapes和CamVid最高分别可达到74.1％和70.1％的交并比(mIoU).DALNet在输入图像分辨率为512×1024的情况下,基于GTX 1080Ti GPU可以获得74 frame/s的前向推理速度,远超实时语义分割所需的速度要求.

关键词：快速语义分割、轻量化网络、通道注意力、空间注意力、城市交通

所属期刊栏目：16

分类号：TP391.4(计算技术、计算机技术)

资助基金：国家自然科学基金;广东省基础与应用基础研究基金;广东省重点实验室项目;广州市创新领军人才项目;暨南大学中央高校基本科研业务费项目;暨大-泰斗联合培养研究生基地项目

在线出版日期：2022-10-21（万方平台首次上网日期，不代表论文的发表时间）

页数：共10页

页码：2377-2386