Traffic-Sign-Detection Algorithm Based on SK-EVC-YOLO

Faguo Zhou (), Huichang Zu (), Yang Li, Yanan Song, Junbin Liao and Changshuo Zheng
Additional contact information
Faguo Zhou: School of Artificial Intelligence, China University of Mining and Technology-Beijing, Beijing 100083, China
Huichang Zu: School of Artificial Intelligence, China University of Mining and Technology-Beijing, Beijing 100083, China
Yang Li: School of Artificial Intelligence, China University of Mining and Technology-Beijing, Beijing 100083, China
Yanan Song: School of Artificial Intelligence, China University of Mining and Technology-Beijing, Beijing 100083, China
Junbin Liao: School of Artificial Intelligence, China University of Mining and Technology-Beijing, Beijing 100083, China
Changshuo Zheng: School of Artificial Intelligence, China University of Mining and Technology-Beijing, Beijing 100083, China

Mathematics, 2023, vol. 11, issue 18, 1-12

Abstract: Traffic sign detection is an important research direction in the process of intelligent transportation in the Internet era, and plays a crucial role in ensuring traffic safety. The purpose of this research is to propose a traffic-sign-detection algorithm based on the selective kernel attention (SK attention), explicit visual center (EVC), and YOLOv5 model to address the problems of small targets, incomplete detection, and insufficient detection accuracy in natural and complex road situations. First, the feature map with a smaller receptive field in the backbone network is fused with other scale feature maps to increase the small target detection layer. Then, the SK attention mechanism is introduced to extract and weigh features at different scales and levels, enhancing the attention to the target. By fusing the explicit visual center to gather local area features within the layer, the detection effect of small targets is improved. According to the experiment results, the mean average precision (mAP) on the Tsinghua-Tencent Traffic Sign Dataset (TT100K) for the proposed algorithm is 88.5%, which is 4.6% higher than the original model, demonstrating the practicality of the detection of small traffic signs.

Keywords: traffic sign; YOLOv5; small-target detection; attention mechanism; explicit visual center (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2023
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2227-7390/11/18/3873/pdf (application/pdf)
https://www.mdpi.com/2227-7390/11/18/3873/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jmathe:v:11:y:2023:i:18:p:3873-:d:1237542

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

More articles in Mathematics from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().