Abstract:Traffic light detection and recognition are crucial for enhancing the security of unmanned systems. This study proposes a YOLOv5-based traffic light-detection algorithm to tackle the challenges posed by small targets and complex urban backgrounds. Initially, the Mosaic-9 method is employed to enhance the training dataset, thereby boosting the network's ability to generalize and adapt to real-world scenarios. Furthermore, the Squeeze-and-Excitation (SE) attention mechanism is incorporated to improve the network. Moreover, the YOLOv5 algorithm's loss function is optimized by substituting it with Efficient Intersection over Union loss (EIoU_loss), which addresses issues like missed detection and false alarms. Experimental results demonstrate that the model trained with this enhanced network achieves an mAP (mean average precision) of 99.4% on a custom dataset, which is 6.3% higher than that of the original YOLOv5, while maintaining a detection speed of 74 f/s. Therefore, this algorithm offers higher detection accuracy and effectively meets real-time operational requirements. The proposed method has a strong application potential, and can be widely used in the field of automatic driving, assisted driving, etc. Its application is not only of great significance in improving the accuracy and speed of traffic sign detection, but also can provide technical support for the development of intelligent transportation systems.

Real-time Detection of Vehicle and Traffic Light for Intelligent and Connected Vehicles Based on YOLOv3 Network

Design of Real-Time Vehicle Detection Based on YOLOv4

TLDM: An Enhanced Traffic Light Detection Model Based on YOLOv5

Improved Real-Time Traffic Obstacle Detection and Classification Method Applied in Intelligent and Connected Vehicles in Mixed Traffic Environment

Traffic Light Detection Algorithm Based on Improved YOLOXs

Traffic Light Detection and Recognition Method Based on YOLOv5s and AlexNet

An Enhanced Deep Learning Model for Obstacle and Traffic Light Detection Based on YOLOv5

YOLOv3 Tiny Vehicle: a New Model for Real-Time Vehicle Detection

Light-weighted Vehicle Detection Network Based on Improved YOLOv3-tiny

Road Traffic Vehicle Detection Method Using Lightweight YOLOv5 and Attention Mechanism

Light-YOLOv3: License Plate Detection in Multi-Vehicle Scenario.

Research on vehicle detection based on improved YOLOv8 network

Optimized YOLOv3 Algorithm and Its Application in Traffic Flow Detections

Real-time Detection Algorithm for Non-Motorized Vehicles Based on D-YOLO Model.

Real-Time Vehicle Detection Based on Improved YOLO v5

Traffic Vehicle Detection Algorithm Based on YOLOv3

Traffic Light Detection Based on Optimized YOLOv3 Algorithm

Research on Automatic Driving Target Detection Based on YOLOv5s

Real-time Detection Model of Highway Vehicle Based on YOLOv5s

Embedded System Real-Time Vehicle Detection Based on Improved YOLO Network

Vehicle collaborative detection based on YOLOv5