关键词:
交通标志检测
SSD算法
残差网络
特征融合
CBAM
摘要:
目的 针对自然场景下交通标志检测存在的小目标精度低和目标特征信息不足等问题,提出一种使用残差网络(Residual Network, ResNet)和注意力机制(Attention Mechanism)的单步多目标检测算法SSD(Single Shot MultiBox Detector),经过残差网络和注意力机制提取的特征向量输送到一个轻量、高效的特征融合模块中,最后将输出的feature map送到检测器中进行检测,从而提升交通标志检测的正确率。方法 首先,利用残差模块将特征进行1×1降维再3×3升维,然后将恒等映射和残差部分生成的特征图进行逐像素相加;其次,将CBAM(Convolutional Block Attention Module)引入到残差模块Conv4_x输出的特征图上,然后与残差模块Conv2_x,Conv3_x输出的特征图一起输入到高效的特征融合模块中进行特征融合,最后将融合后的特征图送入模型中检测以实现对交通标志的识别。结果 通过仿真实验验证,改进后的单步多目标检测算法SSD在中国交通标志检测数据集上进行检测的平均精度为90.55%,能够有效地提取小目标特征的信息。相较于主流算法CenterNet、YOLOv3、YOLOv4、Faster R-CNN、SSD分别提高了2.57%、3.4%、2.79%、3.8%、4.93%。结论 优化后的目标检测方法相较于其他检测方法提取到了更多的特征信息,达到了更高的检测精度,在交通标志检测中具有良好的实用性和有效性。