基于改进SSD的交通标志检测算法
Traffic sign detection algorithm based on improved SSD作者机构:长安大学信息工程学院西安710000
出 版 物:《电子测量技术》 (Electronic Measurement Technology)
年 卷 期:2023年第46卷第7期
页 面:151-158页
学科分类:08[工学] 080203[工学-机械设计及理论] 0802[工学-机械工程]
主 题:交通标志检测 SSD ResNest K-means++ RFB模块 加权特征融合
摘 要:为了解决真实交通场景下交通标志因目标较小而导致检测精度低的问题,提出了一种改进SSD的交通标志检测算法。首先使用更深层次的ResNest网络替换原始SSD算法的主干网络VGG16来增强弱目标特征的强表征能力,然后在SSD的额外添加层使用RFB模块来增加小目标的感受野。其次使用Bi-FPN加权双向特征金字塔网络有效结合深层与浅层的特征信息,改善小目标的检测性能。最后使用K-means++聚类算法调整默认窗口的大小,有效避免因原始默认窗口太大但交通标志较小而无法匹配的问题,以改善检测效率。实验结果表明,本文提出的模型在中国交通标志数据集(CCTSDB)上获得了95.33%的mAP,与原始SSD模型相比,本文所构建的模型能更好的适应自然背景下的交通标志检测。