Optimizing YOLOv4 Hyperparameters for Enhanced Vehicle Detection in Intelligent Transportation Systems

被引:0
作者
Deepak, G. Divya [1 ]
Bhat, Subraya Krishna [1 ]
机构
[1] Manipal Inst Technol, Manipal Acad Higher Educ, Dept Mech & Ind Engn, Manipal 576104, Karnataka, India
关键词
Vehicle detection; Vehicle monitoring; CNN; YOLO; YOLOv4;
D O I
10.1007/s13177-025-00519-3
中图分类号
U [交通运输];
学科分类号
08 ; 0823 ;
摘要
Optimizing the hyperparameters of object detection models is critical for enhancing their performance, particularly towards domain-specific tasks such as vehicle detection. This work systematically investigates the optimization of key hyperparameters for the YOLOv4 model to maximize its efficiency in detecting vehicles. We evaluated the impact of two base CNN architectures (Tiny and CSP-DarkNet53), solvers (SGDM, RMSProp, Adam), learning rates (10- 5, 10- 4, 10- 3), and detection thresholds (0.1, 0.2, 0.3) on the model performance. Our findings reveal that the optimal performance, achieving an average precision-recall (PR (avg)) value of 99%, is obtained using CSP-DarkNet53 network with a learning rate of 10- 5 and a detection threshold of 0.1, employing the Adam solver across all learning rates and detection thresholds studied herein. Decreasing the learning rate from 10- 3 to 10- 5 and detection thresholds from 0.3 to 0.1, steadily enhanced network performance. Additionally, the choice of solver significantly influences the model's performance. The results emphasize the importance of meticulous hyperparameter tuning to improve the accuracy of object detection models. By leveraging the optimized YOLOv4 network, traffic management authorities and autonomous vehicle developers can benefit from a more reliable and efficient vehicle monitoring solution. This study contributes to the broader field of intelligent transportation systems, offering a pathway to practical implementations of object detection technologies in real-world applications.
引用
收藏
页数:13
相关论文
共 50 条
[1]   Vehicle detection systems for intelligent driving using deep convolutional neural networks [J].
Abiyev R. ;
Arslan M. .
Discover Artificial Intelligence, 2023, 3 (01)
[2]  
Ahn K, 2022, PR MACH LEARN RES, P247
[3]  
Amit Y., 2021, Computer Vision: A Reference Guide, P875, DOI [10.1007/978-3-030-63416-2, DOI 10.1007/978-3-030-63416-2, 10.1007/978-3-030-63416-2660, DOI 10.1007/978-3-030-63416-2660]
[4]  
Beger A., 2016, Precision-Recall Curves, DOI DOI 10.2139/SSRN.2765419
[5]  
Bochkovskiy A, 2020, Arxiv, DOI [arXiv:2004.10934, DOI 10.48550/ARXIV.2004.10934]
[6]  
Chen Aoxiang, 2025, ITM Web of Conferences, V70, DOI 10.1051/itmconf/20257003008
[7]   Occlusion and multi-scale pedestrian detection A review [J].
Chen, Wei ;
Zhu, Yuxuan ;
Tian, Zijian ;
Zhang, Fan ;
Yao, Minda .
ARRAY, 2023, 19
[8]   A comparative study of breast tumour detection using a semantic segmentation network coupled with different pretrained CNNs [J].
Deepak, G. Divya ;
Bhat, Subraya Krishna .
COMPUTER METHODS IN BIOMECHANICS AND BIOMEDICAL ENGINEERING-IMAGING AND VISUALIZATION, 2024, 12 (01)
[9]   Deep learning-based CNN for multiclassification of ocular diseases using transfer learning [J].
Deepak, G. Divya ;
Bhat, Subraya Krishna .
COMPUTER METHODS IN BIOMECHANICS AND BIOMEDICAL ENGINEERING-IMAGING AND VISUALIZATION, 2024, 12 (01)
[10]   Optimization of deep neural networks for multiclassification of dental X-rays using transfer learning [J].
Deepak, G. Divya ;
Bhat, Subraya Krishna .
COMPUTER METHODS IN BIOMECHANICS AND BIOMEDICAL ENGINEERING-IMAGING AND VISUALIZATION, 2023, 12 (01)