<p>With advancements in computer vision and artificial intelligence, traffic sign recognition systems have become essential in advanced driver assistance and autonomous driving systems. These systems enable the precise detection of key road information. However, recognizing small traffic signs in real-world scenarios remains a significant challenge due to their limited size and features. In this study, we propose YOLO-TSR, an efficient approach for detecting small traffic signs, inspired by the YOLOv8 framework. This method offers three major contributions: (1) We introduce an efficient attention mechanism in the Backbone to enhance focus on small targets; (2) We propose a downsampling process using slicing and reassembling operations in the backbone, which preserve information and improve feature extraction for small objects; (3) We refine the upsampling process in the head by applying the content-aware CARAFE operation, which enhances the model’s detection performance. Experiments on the challenging TT100K and CCTSDB2021 datasets show that YOLO-TSR achieves a mAP50 of 72.73% and a mAP50-95 of 56.57% on TT100K, and a mAP50 of 87.86% and a mAP50-95 of 57.78% on CCTSDB2021, surpassing the performance of the original YOLOv8n on both datasets. Additionally, this method is real-time and demonstrates great potential for applications in advanced driver assistance systems and autonomous driving systems.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Enhancing small traffic sign recognition based on an improved YOLOv8 algorithm

  • Xiaoqing Zhang,
  • Weixi Wang,
  • Qi Wang,
  • Liang Shan,
  • Satoshi Yamane,
  • Bochao Zheng,
  • Sifan Liu,
  • Tianxing Shen,
  • Dongming Ge

摘要

With advancements in computer vision and artificial intelligence, traffic sign recognition systems have become essential in advanced driver assistance and autonomous driving systems. These systems enable the precise detection of key road information. However, recognizing small traffic signs in real-world scenarios remains a significant challenge due to their limited size and features. In this study, we propose YOLO-TSR, an efficient approach for detecting small traffic signs, inspired by the YOLOv8 framework. This method offers three major contributions: (1) We introduce an efficient attention mechanism in the Backbone to enhance focus on small targets; (2) We propose a downsampling process using slicing and reassembling operations in the backbone, which preserve information and improve feature extraction for small objects; (3) We refine the upsampling process in the head by applying the content-aware CARAFE operation, which enhances the model’s detection performance. Experiments on the challenging TT100K and CCTSDB2021 datasets show that YOLO-TSR achieves a mAP50 of 72.73% and a mAP50-95 of 56.57% on TT100K, and a mAP50 of 87.86% and a mAP50-95 of 57.78% on CCTSDB2021, surpassing the performance of the original YOLOv8n on both datasets. Additionally, this method is real-time and demonstrates great potential for applications in advanced driver assistance systems and autonomous driving systems.