STDE-YOLOv5s: a traffic sign detection algorithm based on context information enhancement
Abstract With the rapid development of autonomous driving technology, traffic sign detection has become a crucial component. Existing object detection networks encounter difficulties in accurately identifying small sized traffic signs. Considering the strong capability of the YOLOv5s algorithm in sm...
Saved in:
| Main Authors: | , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Springer
2025-07-01
|
| Series: | Discover Artificial Intelligence |
| Subjects: | |
| Online Access: | https://doi.org/10.1007/s44163-025-00405-7 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1849764302691500032 |
|---|---|
| author | Qi Tian Juwei Zhang |
| author_facet | Qi Tian Juwei Zhang |
| author_sort | Qi Tian |
| collection | DOAJ |
| description | Abstract With the rapid development of autonomous driving technology, traffic sign detection has become a crucial component. Existing object detection networks encounter difficulties in accurately identifying small sized traffic signs. Considering the strong capability of the YOLOv5s algorithm in small object detection, its relatively low number of parameters, and ease of deployment, this paper selects YOLOv5s as the base model and proposes an improved traffic sign detection algorithm. Firstly, the feature fusion network is improved. By adding horizontal connection paths and a weighted fusion mechanism, the efficiency of multi scale feature fusion is enhanced. Additionally, by incorporating more backbone feature information into the Neck layer, the loss of detailed information is reduced, making small sized traffic signs more distinct in the images. Finally, a Context Information Enhanced Module that combines dilated convolution and deformable convolution is proposed, which enhances the network’s ability to extract context information. The proposed improved algorithm has achieved excellent results on the TT100K dataset. Its mAP50 outperforms existing advanced models such as YOLOv8 and YOLOv11. |
| format | Article |
| id | doaj-art-710ee07f40af497e992fb0662c097c47 |
| institution | DOAJ |
| issn | 2731-0809 |
| language | English |
| publishDate | 2025-07-01 |
| publisher | Springer |
| record_format | Article |
| series | Discover Artificial Intelligence |
| spelling | doaj-art-710ee07f40af497e992fb0662c097c472025-08-20T03:05:10ZengSpringerDiscover Artificial Intelligence2731-08092025-07-015111810.1007/s44163-025-00405-7STDE-YOLOv5s: a traffic sign detection algorithm based on context information enhancementQi Tian0Juwei Zhang1College of Information Engineering and Artificial Intelligence, Henan University of Science and TechnologyCollege of Information Engineering and Artificial Intelligence, Henan University of Science and TechnologyAbstract With the rapid development of autonomous driving technology, traffic sign detection has become a crucial component. Existing object detection networks encounter difficulties in accurately identifying small sized traffic signs. Considering the strong capability of the YOLOv5s algorithm in small object detection, its relatively low number of parameters, and ease of deployment, this paper selects YOLOv5s as the base model and proposes an improved traffic sign detection algorithm. Firstly, the feature fusion network is improved. By adding horizontal connection paths and a weighted fusion mechanism, the efficiency of multi scale feature fusion is enhanced. Additionally, by incorporating more backbone feature information into the Neck layer, the loss of detailed information is reduced, making small sized traffic signs more distinct in the images. Finally, a Context Information Enhanced Module that combines dilated convolution and deformable convolution is proposed, which enhances the network’s ability to extract context information. The proposed improved algorithm has achieved excellent results on the TT100K dataset. Its mAP50 outperforms existing advanced models such as YOLOv8 and YOLOv11.https://doi.org/10.1007/s44163-025-00405-7Small target detectionYOLOv5sTraffic sign detectionDeep learning |
| spellingShingle | Qi Tian Juwei Zhang STDE-YOLOv5s: a traffic sign detection algorithm based on context information enhancement Discover Artificial Intelligence Small target detection YOLOv5s Traffic sign detection Deep learning |
| title | STDE-YOLOv5s: a traffic sign detection algorithm based on context information enhancement |
| title_full | STDE-YOLOv5s: a traffic sign detection algorithm based on context information enhancement |
| title_fullStr | STDE-YOLOv5s: a traffic sign detection algorithm based on context information enhancement |
| title_full_unstemmed | STDE-YOLOv5s: a traffic sign detection algorithm based on context information enhancement |
| title_short | STDE-YOLOv5s: a traffic sign detection algorithm based on context information enhancement |
| title_sort | stde yolov5s a traffic sign detection algorithm based on context information enhancement |
| topic | Small target detection YOLOv5s Traffic sign detection Deep learning |
| url | https://doi.org/10.1007/s44163-025-00405-7 |
| work_keys_str_mv | AT qitian stdeyolov5satrafficsigndetectionalgorithmbasedoncontextinformationenhancement AT juweizhang stdeyolov5satrafficsigndetectionalgorithmbasedoncontextinformationenhancement |