CTM-YOLOv8n: A Lightweight Pedestrian Traffic-Sign Detection and Recognition Model with Advanced Optimization

被引:1
|
作者
Chen, Qiang [1 ,2 ]
Dai, Zhongmou [1 ]
Xu, Yi [3 ,4 ]
Gao, Yuezhen [5 ]
机构
[1] Tianjin Univ Technol & Educ, Sch Automobile & Transportat, Tianjin 300222, Peoples R China
[2] Natl & Local Joint Engn Res Ctr Intelligent Vehicl, Tianjin 300222, Peoples R China
[3] Shandong Univ Technol, Sch Transportat & Vehicle Engn, Zibo 255000, Peoples R China
[4] Qingte Grp Co Ltd, Qingdao 266106, Peoples R China
[5] Univ Alberta, Dept Civil Engn, 116 St NW, Edmonton, AB T6G 2E1, Canada
来源
WORLD ELECTRIC VEHICLE JOURNAL | 2024年 / 15卷 / 07期
关键词
traffic-sign detection and recognition; YOLOv8n; C2f Faster; MPDIoU; lightweight;
D O I
10.3390/wevj15070285
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Traffic-sign detection and recognition (TSDR) is crucial to avoiding harm to pedestrians, especially children, from intelligent connected vehicles and has become a research hotspot. However, due to motion blurring, partial occlusion, and smaller sign sizes, pedestrian TSDR faces increasingly significant challenges. To overcome these difficulties, a CTM-YOLOv8n model is proposed based on the YOLOv8n model. With the aim of extracting spatial features more efficiently and making the network faster, the C2f Faster module is constructed to replace the C2f module in the head, which applies filters to only a few input channels while leaving the remaining ones untouched. To enhance small-sign detection, a tiny-object-detection (TOD) layer is designed and added to the first C2f layer in the backbone. Meanwhile, the seventh Conv layer, eighth C2f layer, and connected detection head are deleted to reduce the quantity of model parameters. Eventually, the original CIoU is replaced by the MPDIoU, which is better for training deep models. During experiments, the dataset is augmented, which contains the choice of categories 'w55' and 'w57' in the TT100K dataset and a collection of two types of traffic signs around the schools in Tianjin. Empirical results demonstrate the efficacy of our model, showing enhancements of 5.2% in precision, 10.8% in recall, 7.0% in F1 score, and 4.8% in mAP@0.50. However, the number of parameters is reduced to 0.89M, which is only 30% of the YOLOv8n model. Furthermore, the proposed CTM-YOLOv8n model shows superior performance when tested against other advanced TSDR models.
引用
收藏
页数:19
相关论文
共 29 条
  • [1] A Lightweight Traffic Sign Recognition Model Based on Improved YOLOv5
    Yang, Jie
    Sun, Ting
    Zhu, Wenchao
    Li, Zonghao
    IEEE ACCESS, 2023, 11 : 115998 - 116010
  • [2] A Novel Lightweight Traffic Sign Recognition Model Based on YOLOv5
    Li, Wenju
    Zhang, Gan
    Cui, Liu
    JOURNAL OF TRANSPORTATION ENGINEERING PART A-SYSTEMS, 2023, 149 (05)
  • [3] Detection of Traffic Signs Based on Lightweight YOLOv8n
    Liu, Shihong
    Li, Shiwei
    2024 5TH INTERNATIONAL CONFERENCE ON COMPUTER ENGINEERING AND APPLICATION, ICCEA 2024, 2024, : 1200 - 1204
  • [4] Tp-yolov8: a lightweight and accurate model for traffic accident recognition
    Ning, Zhaole
    Zhang, Tianze
    Li, Xin
    Wu, Aiying
    Shi, Gang
    JOURNAL OF SUPERCOMPUTING, 2025, 81 (04):
  • [5] Lightweight Traffic Sign Recognition and Detection Algorithm Based on Improved YOLOv5s
    Liu, Fei
    Zhong, Yanfen
    Qiu, Jiawei
    LASER & OPTOELECTRONICS PROGRESS, 2024, 61 (24)
  • [6] A Lightweight Model for Road Sign Detection Based on Improved-YOLOv8
    Wang, Bingxuan
    Wang, Gengzhao
    2024 9TH INTERNATIONAL CONFERENCE ON ELECTRONIC TECHNOLOGY AND INFORMATION SCIENCE, ICETIS 2024, 2024, : 319 - 322
  • [7] Traffic Sign Detection Algorithm Based on Improved YOLOv8n
    Peng, Jun
    Mou, Biao
    Jin, Shangzhu
    Lu, Yiyi
    Li, Chenxi
    Chen, Wei
    Jiang, Aiping
    2024 IEEE 19TH CONFERENCE ON INDUSTRIAL ELECTRONICS AND APPLICATIONS, ICIEA 2024, 2024,
  • [8] SEDG-Yolov5: A Lightweight Traffic Sign Detection Model Based on Knowledge Distillation
    Zhao, Liang
    Wei, Zhengjie
    Li, Yanting
    Jin, Junwei
    Li, Xuan
    ELECTRONICS, 2023, 12 (02)
  • [9] An Improved Traffic Sign Detection and Recognition Deep Model Based on YOLOv5
    Wang, Qianying
    Li, Xiangyu
    Lu, Ming
    IEEE ACCESS, 2023, 11 : 54679 - 54691
  • [10] RDB-YOLOv8n: Insulator defect detection based on improved lightweight YOLOv8n model
    Jiang, Yong
    Wang, Shuai
    Cao, Weifeng
    Liang, Wanyong
    Shi, Jun
    Zhou, Lintao
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2024, 21 (05)