Research Article(20250301)
Performance Analysis of YOLOv4 Tiny for Real-Time Traffic Sign Recognition on Edge AI Systems Under Adverse Conditions
Nitesh Pandey*, Chandrapal Singh Umath, Amit Kumar Srivastava
Department of Computer Science and Engineering, Student, Jaypee University of Engineering and Technology, Raghogarh Guna (M.P.) - 473226, India
*Corresponding author’s Email: niteshpandey46974@gmail.com
Pages 1-14 | Received 11 Jan 2025, Accepted 22 Feb 2025, Published online: 01 March 2025
Abstract
Traffic sign detection and recognition are two important capabilities of intelligent transportation systems or autonomous vehicles that can be used to feature real-time traffic sign detection. This paper is a proposal for real time lightweight traffic sign detection system running on Raspberry Pi 4 Model B with employment of YOLOv4-Tiny for its object classification and detection. Training was conducted on German Traffic Sign Recognition Benchmark (GTSRB) dataset which has 50,000+ images of different traffic signs in total of 43 classes. Image augmentation, contrast normalization and adaptive thresholding techniques were used to enhance the detection performance with different illumination on the image. We train the model on Darknet for 100 epochs with a batch size of 64 and achieve 96.5% test set accuracy as well as a mAP of 94.2%. Using OpenVINO and TensorRT, we optimized the model to be used for real time inference on Raspberry Pi that was achieved with 45ms per frame latency. Performance evaluations revealed that the system could consistently determine and classify traffic signs in the daylight, low light, and occlusion cases with 2.3% false positives and 3.7% false negatives. Aspect input speed of our YOLO model in comparison with conventional machine learning methods is 32% faster (HOG SVM), 110% faster (haarcascades), 21% more accurate (HOG SVM), and 18% more accurate (haar cascades). We have deployed it on edge devices, such as Raspberry Pi, which makes it suitable to be used in real world applications of intelligent cars, road surveillance, and driver assistance systems. Future work toward decreasing power consumption, adding multi class objects tracking and increasing robustness to incorrect motion blur and bad weather will be addressed.
Keywords
Traffic Sign Detection; YOLOv4-Tiny; Deep Learning; Convolutional Neural Networks (CNN); Object Detection; Real-time Image Processing; Machine Learning; Computer Vision