Traffic sign detection is an important issue in the fields of automatic driving. Aiming at the problem of YOLOv5 algorithm in detection small target traffic signs, an algorithm based on improved YOLOv5 is proposed. First, the Coordinate Attention (CA) model was integrated into the backbone back-end to capture the relationship between location information and channels, so as to obtain the region of interest more accurately. Then, the upsampling method in the feature fusion network was changed to transposed convolution to improve the learning ability of the network and the detection accuracy. Finally, cross-layer connections were introduced in the feature fusion network to fuse more shallow feature information, thereby improving the feature extraction ability of the network and further improving the detection accuracy of small objects. The experimental results show that the improved algorithm achieves 89.73% mAP on the traffic sign detection data set TT100K, which is 7.1% higher than the original YOLOv5. Therefore, the proposed improved YOLOv5 algorithm can effectively improve the accuracy of traffic sign detection.
Traffic Sign Detection Based on Improved YOLOv5 Model
Lect. Notes Electrical Eng.
International Conference on Green Intelligent Transportation System and Safety ; 2022 ; Qinghuangdao, China September 16, 2022 - September 18, 2022
31.12.2024
15 pages
Aufsatz/Kapitel (Buch)
Elektronische Ressource
Englisch
Traffic sign recognition based on YOLOv5
SPIE | 2022
|Research and Implementation of an Embedded Traffic Sign Detection Model Using Improved YOLOV5
Springer Verlag | 2024
|Research and Implementation of an Embedded Traffic Sign Detection Model Using Improved YOLOV5
Springer Verlag | 2024
|