一种融合注意力机制与上下文信息的交通标志检测方法

A Traffic Sign Detection Based on Attentional Mechanism and Contextual Information

针对当前交通标志检测中存在小目标检测精度低、检测实时性不高以及目标漏检等问题,在YOLOv3基础上提出了一种融合了注意力机制与上下文信息的交通标志检测方法;首先通过改进通道注意力机制的压缩方式,对特征图通道重新进行标定;然后引入空间金字塔池化模块SPP;最后增加特征映射并拼接到原特征融合网络中的小目标部分,充分利用上下文信息增强对小目标的检测;实验结果表TT100K(Tsinghua-Tencent 100K)交通标志数据集上,与YOLOv3网络相比,在每秒传输帧数(FPS,frame per second)变化不大的情况下,平均精度均值和小目标的精度均值分别提升3.03%和4.59%;实验结果证明了改进网络在小目标检测和整体检测中的有效性。

Aiming at the problems of low accuracy for small targets,low real-time detection and missed target detection in current traffic sign detection,a traffic sign detection method that combines attention mechanism and context information is proposed on the basis of YOLOv3.,In this method,firstly,the channel of feature map is recalibrated by improving the compression method of channel attention mechanism;then the spatial pyramid pooling module SPP is introduced;finally,the feature mapping is added and spliced into the small target part of the original feature fusion network,making full use of contextual information to improve the detection of small targets.Compared with the original YOLOv3 network,the experimental results on the TT100 K(Tsinghua-Tencent 100 K) traffic sign dataset show that the mean average precision and the mean average precision on small target increases respectively by 3.03% and 4.59% with little change in frames per second(FPS).The experimental results demonstrate the effectiveness of the improved network in small target detection and overall detection.