融合前景注意力的轻量级交通标志检测网络

作者：俞林森; 陈志国^*

来源：电子测量与仪器学报, 2023, 37(01): 21-31.

DOI：10.13382/j.jemi.B2205524

• 交通标志检测
• 激活函数
• 前景注意力
• 特征融合
• VariFocalLoss
• GIoU

针对目标检测算法模型在交通标志检测上容易出现错检和漏检等问题，提出一种融合前景注意力的轻量级交通标志检测网络YOLOT。首先引入SiLU激活函数，提升模型检测的准确率；其次设计了一种基于鬼影模块的轻量级骨干网络，有效提取目标物特征；接着引入前景注意力感知模块，抑制背景噪声；然后改进路径聚合网络，加入残差结构，充分学习底层特征信息；最后使用VariFocalLoss和GIoU,分别计算目标的分类损失和目标间的相似度，使目标的分类和定位更加准确。在多个数据集上进行了大量实验，结果表明，本文方法的精度优于目前最先进方法，在CCTSDB数据集上进行消融实验，最终精度达到98.50%,与基线模型相比，准确率提升1.32%,同时模型仅4.7 MB,实时检测帧率达到44 FPS。

• 单位
  江南大学