LLTH-YOLOv5: A Real-Time Traffic Sign Detection Algorithm for Low-Light Scenes

Traffic sign detection is a crucial task for autonomous driving systems. However, the performance of deep learning-based algorithms for traffic sign detection is highly affected by the illumination conditions of scenarios. While existing algorithms demonstrate high accuracy in well-lit environments,...

Celý popis

Uloženo v:
Podrobná bibliografie
Vydáno v:Automotive innovation (Online) Ročník 7; číslo 1; s. 121 - 137
Hlavní autoři: Sun, Xiaoqiang, Liu, Kuankuan, Chen, Long, Cai, Yingfeng, Wang, Hai
Médium: Journal Article
Jazyk:angličtina
Vydáno: Singapore Springer Nature Singapore 01.02.2024
Témata:
Automotive Engineering
Control
Electrical Engineering
Energy Efficiency
Engineering
Mechatronics
Robotics
Deep learning
YOLOv5
Traffic sign detection
Low-light enhancement
Object detection
ISSN:2096-4250, 2522-8765
On-line přístup:Získat plný text
Tagy: Přidat tag
Žádné tagy, Buďte první, kdo vytvoří štítek k tomuto záznamu!
Popis
Shrnutí:Traffic sign detection is a crucial task for autonomous driving systems. However, the performance of deep learning-based algorithms for traffic sign detection is highly affected by the illumination conditions of scenarios. While existing algorithms demonstrate high accuracy in well-lit environments, they suffer from low accuracy in low-light scenarios. This paper proposes an end-to-end framework, LLTH-YOLOv5, specifically tailored for traffic sign detection in low-light scenarios, which enhances the input images to improve the detection performance. The proposed framework comproses two stages: the low-light enhancement stage and the object detection stage. In the low-light enhancement stage, a lightweight low-light enhancement network is designed, which uses multiple non-reference loss functions for parameter learning, and enhances the image by pixel-level adjustment of the input image with high-order curves. In the object detection stage, BIFPN is introduced to replace the PANet of YOLOv5, while designing a transformer-based detection head to improve the accuracy of small target detection. Moreover, GhostDarkNet53 is utilized based on Ghost module to replace the backbone network of YOLOv5, thereby improving the real-time performance of the model. The experimental results show that the proposed method significantly improves the accuracy of traffic sign detection in low-light scenarios, while satisfying the real-time requirements of autonomous driving.
ISSN:2096-4250
2522-8765
DOI:10.1007/s42154-023-00249-w