Enhancing autonomous driving safety: A robust traffic sign detection and recognition model TSD-YOLO

Abstract

As autonomous driving technology rapidly advances, Traffic Sign Detection and Recognition (TSDR) has become pivotal in ensuring the safety and regulatory compliance of autonomous vehicles. Despite progress, existing technologies struggle under challenging conditions such as adverse weather and complex roadway environments. To overcome these obstacles, we introduce a novel model, TSD-YOLO, which leverages Mamba and YOLO technologies to enhance the accuracy and robustness of traffic sign detection. Our innovative YOLO-MAM dual-branch module merges convolutional layer-based local feature extraction with the long-distance dependency capabilities of the State Space Model (SSM). We conducted experimental validations using the Tsinghua-Tencent 100K (TT-100K) dataset and the Mapillary Traffic Sign Detection (MTSD) dataset, demonstrating our model’s efficacy across various datasets. Furthermore, cross-dataset validations affirm the model’s exceptional generalization and robustness across diverse environments. This study not only bolsters traffic sign detection and recognition in autonomous driving systems but also paves the way for future advancements in autonomous driving technology.