摘要:目的 针对自然场景下交通标志检测存在的小目标精度低和目标特征信息不足等问题,提出一种使用残差 网络( 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%。 结论 优化后的目标检测方法相较于其他检测方法提 取到了更多的特征信息,达到了更高的检测精度,在交通标志检测中具有良好的实用性和有效性。