Preprint Article Version 1 This version is not peer-reviewed

YOLO-ADual: Lightweight Traffic Sign Detection Model on Mobile Driving System

Version 1 : Received: 12 July 2024 / Approved: 13 July 2024 / Online: 15 July 2024 (10:19:30 CEST)

How to cite: Fang, S.; Chen, C.; Li, Z.; Zhou, M.; Wei, R. YOLO-ADual: Lightweight Traffic Sign Detection Model on Mobile Driving System. Preprints 2024, 2024071126. https://doi.org/10.20944/preprints202407.1126.v1 Fang, S.; Chen, C.; Li, Z.; Zhou, M.; Wei, R. YOLO-ADual: Lightweight Traffic Sign Detection Model on Mobile Driving System. Preprints 2024, 2024071126. https://doi.org/10.20944/preprints202407.1126.v1

Abstract

Traffic sign detection plays a pivotal role in autonomous driving systems. The intricacy of the detection model necessitates high-performance hardware. Real-world traffic environments exhibit considerable variability and diversity, posing challenges for effective feature extraction by the model. Therefore, it is imperative to develop a detection model that is not only highly accurate but also lightweight. In this paper, we proposed YOLO-ADual, a novel lightweight model. Our method leverages the C3Dual and Adown lightweight modules as replacements for CPS and CBL modules in YOLOv5. The Adown module effectively mitigates feature loss during downsampling while reducing computational costs. Meanwhile, C3Dual optimizes the processing power for kernel feature extraction, enhancing computation efficiency while preserving network depth and feature extraction capability. Furthermore, the inclusion of the CBAM module enables the network to focus on salient information within the image, thus augmenting its feature representation capability. Our proposed algorithm achieves a [email protected] of 70.1% while significantly reducing the number of parameters and computational requirements to 51.83% and 64.73% of the original model, respectively. Compared to various lightweight models, our approach demonstrates competitive performance in terms of both computational efficiency and accuracy.

Keywords

traffic sign detection; small object detection; YOLO-ADual; attention mechanism; dual convolution

Subject

Computer Science and Mathematics, Computer Vision and Graphics

Comments (0)

We encourage comments and feedback from a broad range of readers. See criteria for comments and our Diversity statement.

Leave a public comment
Send a private comment to the author(s)
* All users must log in before leaving a comment
Views 0
Downloads 0
Comments 0
Metrics 0


×
Alerts
Notify me about updates to this article or when a peer-reviewed version is published.
We use cookies on our website to ensure you get the best experience.
Read more about our cookies here.