|国家科技期刊平台
首页|期刊导航|西安电子科技大学学报|融合全局和局部信息的实时烟雾分割算法

融合全局和局部信息的实时烟雾分割算法OA北大核心CSTPCD

中文摘要

针对烟雾形状不规则、呈半透明状且边界模糊导致烟雾分割困难的问题,提出一种融合全局和局部信息的双分支实时烟雾分割算法。该算法设计了轻量级的Transformer分支和卷积神经网络分支分别提取烟雾的全局特征和局部特征,Transformer分支和卷积神经网络分支共同作用,可以在充分学习烟雾的长距离像素依赖关系的同时保留烟雾细节信息,从而准确区分烟雾和背景像素,改善烟雾分割效果。同时该结构可以满足实际烟雾检测任务的实时性要求;基于多层感知机的解码器充分利用不同尺度的烟雾特征图,并进一步建模烟雾全局上下文信息,增强模型对多尺度烟雾的感知能力,从而提升烟雾分割精度;而且解码器结构简单,可以降低解码器部分的计算量。该算法在自建烟雾分割数据集上的平均交并比为92.88%,模型参数量为2.96 M,推理速度为56.94帧/s。该算法在公开数据集上的综合性能优于其他烟雾检测算法。实验结果表明,该算法分割烟雾的准确率高,推理速度快,可以满足实际烟雾检测任务的准确性和实时性需求。

张欣雨;梁煜;张为;

天津大学微电子学院,天津300072

计算机与自动化

烟雾分割Transformer卷积神经网络双分支

《西安电子科技大学学报》 2024 (001)

P.147-156 / 10

天津市新一代人工智能科技重大专项(19ZXZNGX00030)。

10.19665/j.issn1001-2400.20230405

评论