Recognition of Front-Vehicle Taillights Based on YOLOv5s

In automatic driving, the recognition of Front-Vehicle taillights plays a key role in predicting the intentions of the vehicle ahead. In order to accurately identify the Front-Vehicle taillights, we first analyze the different characteristics of the vehicle taillight signal, and then propose an improved taillight recognition model based on YOLOv5s. First, CA(coordinate attention) is inserted into the backbone network of YOLOv5s model to improve small target recognition and reduce interference from other light sources; Then, the EIOU Loss is used to solve the class imbalance problem; Finally, EIOU-NMS is used to solve the problem of anchor box error suppression in the recognition process. We use the actual scene video and vehicle taillights dataset to conduct ablation experiments to verify the effectiveness of the improved algorithm. The experimental results show that the mAP value of the model is 9.2% higher than YOLOv5s.