Traffic sign recognition is crucial for intelligent transportation and autonomous driving, ensuring road safety and efficient traffic management. This paper proposes a lightweight enhanced MobileViT model (E-MobileViT). It is based on the MobileViT model, combining the advantages of CNN and Transformer. We integrate Efficient Local Attention (ELA) and Convolutional Block Attention Module (CBAM) mechanisms in the model to improve feature extraction. The proposed model improves the feature fusion structure and significantly reduces the number of model parameters. We evaluated the model on the German Traffic Sign Recognition Benchmark (GTSRB), Belgian Traffic Signs Database (BTSD), and China Traffic Signs Database (TSRD) datasets and its accuracy reaches 99.61%, 99.26% and 97.34%, respectively, which outperforms traditional and advanced models. We confirmed the key role of ELA and CBAM mechanisms through ablation experiments. With fewer parameters than mainstream models, our E-MobileViT model is suitable for resource-constrained environments such as mobile devices, providing a balanced solution for traffic sign recognition tasks.
E-MobileViT: a lightweight model for traffic sign recognition
Ind. Artif. Intell.
17.03.2025
Aufsatz (Zeitschrift)
Elektronische Ressource
Englisch
MobileViT , Traffic sign recognition , Attention mechanisms , Transformer Information and Computing Sciences , Information Systems , Computer Science , Artificial Intelligence , Mathematical Modeling and Industrial Mathematics , Popular Computer Science , Computational Intelligence , Robotics and Automation , Industrial and Production Engineering
E-MobileViT: a lightweight model for traffic sign recognition
Springer Verlag | 2025
|