Traffic Light Recognition Based on Improved YOLOv5l

Abstract

Accurate recognition of traffic lights is essential for ensuring the safety of passengers and pedestrians, especially in the context of self-driving car technology. However, traffic lights present challenges due to their small size and limited recognition accuracy. This paper proposes an enhanced version of the YOLOv5l algorithm specifically designed for traffic light recognition. First, the K-means++ clustering algorithm is employed to generate the prior frame. Second, the SiLU activation function in the basic convolution module is replaced with the adaptive Meta-ACONC activation function, significantly improving the model’s detection accuracy. Third, the coordinate attention mechanism is integrated into the trunk feature extraction network to incorporate coordinate information into the channel, thereby enhancing the network’s sensitivity to small target positions and mitigating the ambiguity caused by increased network depth. Finally, the network’s detection scale is improved by removing the original 20 × 20 large target detection head, leading to an improved accuracy and speed for detecting small targets. The proposed approach is evaluated on self-created traffic light datasets, and compared with the original YOLOv5l model; the improved YOLOv5l model achieves a 7.1% increase in mAP@0.5, reaching 83.3%, effectively meeting the requirements for traffic light detection and recognition.

摘要

准确识别交通信号灯对于确保乘客和行人的安全至关重要, 尤其在自动驾驶领域。然而交通信号灯属于小目标, 识别难度大, 识别精度低。针对此问题, 本文提出了一种改进YOLOv5l的交通信号灯识别方法。首先, 引入K-means++聚类算法生成先验框。其次, 将基础卷积模块中的SiLU激活函数替换成自适应激活函数Meta-ACONC, 有效提升了模型的检测精度。然后, 在主干特征提取网络中加入坐标注意力机制, 使坐标信息融入通道之中, 防止随着网络深度的增加而使位置信息越来越模糊, 提升网络对小目标位置信息的敏感度。最后, 对网络检测尺度进行改进, 删除原始20×20的大目标检测头, 提升小目标的检测精度和识别速度。本文所提出的方法在自制的交通信号灯数据集上进行了实验, 改进后的YOLOv5l模型相比于原始YOLOv5l模型, mAP@0.5提高了7.1%, 达到了83.3%, 有效满足了交通信号灯检测和识别的要求。