安伟强,王卫星,闫迪,等.基于道路特征增强改进YOLOv4的模糊交通标志检测[J]. 微电子学与计算机,2023,40(7):73-81. doi: 10.19304/J.ISSN1000-7180.2022.0552
引用本文: 安伟强,王卫星,闫迪,等.基于道路特征增强改进YOLOv4的模糊交通标志检测[J]. 微电子学与计算机,2023,40(7):73-81. doi: 10.19304/J.ISSN1000-7180.2022.0552
AN W Q,WANG W X,YAN D,et al. Improved YOLOv4 based on feature enhancement for vague traffic sign detection[J]. Microelectronics & Computer,2023,40(7):73-81. doi: 10.19304/J.ISSN1000-7180.2022.0552
Citation: AN W Q,WANG W X,YAN D,et al. Improved YOLOv4 based on feature enhancement for vague traffic sign detection[J]. Microelectronics & Computer,2023,40(7):73-81. doi: 10.19304/J.ISSN1000-7180.2022.0552

基于道路特征增强改进YOLOv4的模糊交通标志检测

Improved YOLOv4 based on feature enhancement for vague traffic sign detection

  • 摘要: 由于外界因素干扰容易导致交通标志在图像中成像模糊,极大的降低了交通标志的检测精度. 同时,考虑到交通标志检测的应用场景大多需要较高的实时性,提出一种基于道路特征增强改进YOLOv4的模糊交通标志检测算法. 首先,为了抑制交通标志图片中杂乱的背景信息并有效捕获交通标志特征,在YOLOv4的特征提取部分嵌入了坐标注意力. 然后,YOLOv4采用最大池化进行下采样忽略了移位等变性的丢失,容易导致模糊交通标志的特征提取不稳定. 为了更加有效的提取图像特征,采用BlurPool进行下采样来更加有效的保留模糊交通标志特征. 最后,在特征融合阶段采用DUpsampling进行上采样以建立新插入像素和原有像素之间的相关性. 实验结果表明,改进后的YOLOv4模型尺寸和参数量相较于原算法分别降低了10.18%和11.32%,FPS和mAP分别提升了2.02和1.34. 通过与YOLOv3_SPP、SSD、Faster RCNN和其他交通标志检测算法对比,改进后的YOLOv4性能均优于这些算法. 通过输入图像测试,本文所改进的算法在现实场景中对模糊交通标志检测具有了更好的性能.

     

    Abstract: Due to the interference of external factors, traffic signs are easily blurred in the image, which greatly reduces the detection accuracy of traffic signs. At the same time, considering that most of the application scenarios of traffic sign detection require high real-time performance, a vague traffic sign detection algorithm based on road feature enhancement and improved YOLOv4 is proposed. First, to suppress the cluttered background information in the traffic sign images and effectively capture the traffic sign features, coordinate attention is embedded in the feature extraction part of YOLOv4. Then, YOLOv4 adopts max pooling for downsampling, ignoring the loss of shift and other variability, which easily leads to unstable feature extraction of fuzzy traffic signs. In order to extract image features more effectively, BlurPool is used for downsampling to more effectively retain the features of blurred traffic signs. Finally, DUpsampling is used for upsampling in the feature fusion stage to establish the correlation between the newly inserted pixels and the original pixels. The experimental results show that the size and parameters of the improved YOLOv4 model are reduced by 10.18% and 11.32%, respectively, compared with the original algorithm, and the FPS and mAP are increased by 2.02 and 1.34, respectively. By comparing with YOLOv3_SPP, SSD, Faster RCNN and other traffic sign detection algorithms, the improved YOLOv4 outperforms these algorithms. Through the input image test, the improved algorithm in this paper has better performance for fuzzy traffic sign detection in real scenes.

     

/

返回文章
返回