陈梦涛, 余粟. 基于改进YOLOV4模型的交通标志识别研究[J]. 微电子学与计算机, 2022, 39(1): 17-25. DOI: 10.19304/J.ISSN1000-7180.2021.0858
引用本文: 陈梦涛, 余粟. 基于改进YOLOV4模型的交通标志识别研究[J]. 微电子学与计算机, 2022, 39(1): 17-25. DOI: 10.19304/J.ISSN1000-7180.2021.0858
CHEN Mengtao, YU Su. Research on traffic sign recognition based on improved YOLOV4 model[J]. Microelectronics & Computer, 2022, 39(1): 17-25. DOI: 10.19304/J.ISSN1000-7180.2021.0858
Citation: CHEN Mengtao, YU Su. Research on traffic sign recognition based on improved YOLOV4 model[J]. Microelectronics & Computer, 2022, 39(1): 17-25. DOI: 10.19304/J.ISSN1000-7180.2021.0858

基于改进YOLOV4模型的交通标志识别研究

Research on traffic sign recognition based on improved YOLOV4 model

  • 摘要: 为了解决目前高清街景图像中定位和分类交通标志任务时,出现的图像目标背景复杂,小目标不易识别等一系列问题,提出了一种基于YOLOV4模型改进的交通标志识别新算法.首先,在原骨干网络中嵌入新型注意力机制CA模块,形成一对方向感知和位置敏感的注意力图,使网络能够在更大区域内聚焦有效特征;其次,在颈部特征增强网络处添加适量的增强感受野RFB模块,进一步提升网络的特征融合能力.在TT100K交通标志数据集上进行实验发现,改进算法在IOU阈值为0.5时,相较于原算法的mAP指标提升了近4个百分点,每秒识别帧数达到了39FPS.因此,提出的CA-RFB-YOLOV4算法可以在增加少量计算参数的情况下,显著提升目标数据的识别精度, 更好地满足了实际场景中交通标志识别的需求.

     

    Abstract: At present, there are a series of problems in the current task of Positioning and classification traffic signs in high-definition street view images. The image target background is complicated and the small targets are difficult to recognize. A new traffic sign recognition algorithm based on YOLOV4 model is proposed. Firstly, a new type of attention mechanism CA module is embedded in the original backbone network to form a pair of direction-aware and position-sensitive attention maps, so that the network can focus on effective features in a larger area. Secondly, features in the neck are enhanced Add an appropriate amount of RFB modules to the network to further enhance the feature integration capability of the network. Experiments on the TT100K traffic sign data set found that when the IOU threshold of the improved algorithm is 0.5, the mAP index of the improved algorithm increased by nearly 4%, and the number of detected frames per second reached 39FPS. Therefore, the CA-RFB-YOLOV4 algorithm proposed in this paper can significantly improve the recognition accuracy of target data with a small increase in calculation parameters, and better meet the needs of traffic sign recognition in actual scenes.

     

/

返回文章
返回