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 条
[41]  
Wang ZQ, 2017, CHIN CONTR CONF, P11104, DOI 10.23919/ChiCC.2017.8029130
[42]  
Weber Markus, 2022, CaltechDATA, DOI 10.22002/D1.20084
[43]   Confidence Score: The Forgotten Dimension of Object Detection Performance Evaluation [J].
Wenkel, Simon ;
Alhazmi, Khaled ;
Liiv, Tanel ;
Alrshoud, Saud ;
Simon, Martin .
SENSORS, 2021, 21 (13)
[44]   A review of object detection based on deep learning [J].
Xiao, Youzi ;
Tian, Zhiqiang ;
Yu, Jiachen ;
Zhang, Yinshu ;
Liu, Shuai ;
Du, Shaoyi ;
Lan, Xuguang .
MULTIMEDIA TOOLS AND APPLICATIONS, 2020, 79 (33-34) :23729-23791
[45]   Ship Detection in Optical Remote Sensing Images Using YOLOv4 and Tiny YOLOv4 [J].
Yildirim, Esra ;
Kavzoglu, Taskin .
6TH INTERNATIONAL CONFERENCE ON SMART CITY APPLICATIONS, 2022, 393 :913-924
[46]  
Yu F, 2018, BDD100K DIVERSE DRIV
[47]  
Zhang JZ, 2020, Arxiv, DOI [arXiv:1905.11881, DOI 10.48550/ARXIV.1905.11881]
[48]   Object Detection With Deep Learning: A Review [J].
Zhao, Zhong-Qiu ;
Zheng, Peng ;
Xu, Shou-Tao ;
Wu, Xindong .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2019, 30 (11) :3212-3232
[49]   Moving Object Detection With Deep CNNs [J].
Zhu, Haidi ;
Yan, Xin ;
Tang, Hongying ;
Chang, Yuchao ;
Li, Baoqing ;
Yuan, Xiaobing .
IEEE ACCESS, 2020, 8 :29729-29741
[50]   Object Detection in 20 Years: A Survey [J].
Zou, Zhengxia ;
Chen, Keyan ;
Shi, Zhenwei ;
Guo, Yuhong ;
Ye, Jieping .
PROCEEDINGS OF THE IEEE, 2023, 111 (03) :257-276