改进SSD算法在交通标志检测中的应用

作     者：孙超 温蜜 景俐娜 SUN Chao;WEN Mi;JING Lina

作者机构：上海电力大学计算机科学与技术学院上海200090 

出 版 物：《计算机工程与应用》 (Computer Engineering and Applications)

年 卷 期：2023年第59卷第4期

页      面：147-155页

核心收录：

学科分类：08[工学] 080203[工学-机械设计及理论] 0802[工学-机械工程] 

基　　金：国家自然科学基金(61872230,61802248,61802249,U1936213) 上海市科委项目(20020500600,19511103700) 上海市学术带头人计划(21XD1421500)。 

主　　题：交通标志检测 SSD 特征提取 条件参数卷积 注意力机制 

摘      要：针对交通标志检测存在误检率高、鲁棒性差等问题,提出了一种改进SSD(single shot multibox detector)的交通标志检测方法。首先从不同维度提取交通标志的位置和方向感知信息,改善目标在浅层特征图上的感受野区域。其次使用特例化的卷积内核对深层特征图进行条件参数卷积,增强交通标志的特征表达能力。最后对通道注意力机制进行改进,在特征通道中融入目标空间信息,提升交通标志目标的显著性。实验结果表明,提出的方法相较于原始SSD在CCTSDB数据集上的检测精度提升了7.6个百分点,检测速度达到87.5 FPS;在LISA数据集上的平均准确率为94.6%,检测速率为85.0 FPS。相比于其他的检测方法,改进后的SSD算法在复杂的自然场景中对交通标志具有更好的鲁棒性。