Road vehicle detection based on improved YOLOv3-SPP algorithm

被引:0
作者
Wang T. [1 ,2 ]
Feng H. [1 ,2 ]
Mi R. [3 ]
Li L. [3 ]
He Z. [4 ]
Fu Y. [1 ,2 ]
Wu S. [1 ,2 ]
机构
[1] School of Information and Communication Engineering, Beijing Information Science & Technology University, Beijing
[2] Key Laboratory of Optoelectronic Testing Technology and Instrument, Ministry of Education, Beijing Information Science & Technology University, Beijing
[3] National Computer Network Emergency Response Technical Team, Coordination Center of China, Beijing
[4] Hebei Key Laboratory of Agricultural Big Data, Hebei Agricultural University, Baoding
来源
Tongxin Xuebao/Journal on Communications | 2024年 / 45卷 / 02期
基金
中国国家自然科学基金;
关键词
activation function; atrous convolution; deep learning; vehicle detection; YOLOv3-SPP algorithm;
D O I
10.11959/j.issn.1000-436x.2024046
中图分类号
学科分类号
摘要
Aiming at the problem of low detection accuracy or missing detection caused by dense vehicles and small scale of distant vehicles in the visual detection of urban road scenes, an improved YOLOv3-SPP algorithm was proposed to optimize the activation function and take DIOU-NMS Loss as the boundary frame loss function to enhance the expression ability of the network. In order to improve the feature extraction ability of the proposed algorithm for small targets and occluding targets, the void convolution module was introduced to increase the receptive field of the target. Based on the experimental results, the proposed algorithm improves the mAP by 1.79% when detecting vehicle targets, and also effectively reduce the missing phenomenon when detecting tight vehicle targets. © 2024 Editorial Board of Journal on Communications. All rights reserved.
引用
收藏
页码:68 / 78
页数:10
相关论文
共 27 条
[1]  
LIU Y H., Research on vehicle detection based on deep learning, (2022)
[2]  
JABRI S, SAIDALLAH M, BELRHITI E A A, Et al., Moving vehicle detection using Haar-like, LBP and a machine learning adaboost algorithm, Proceedings of IEEE International Conference on Image Processing, Applications and Systems (IPAS), pp. 121-124, (2018)
[3]  
DAVID H, ATHIRA T A., Improving the performance of vehicle detection and verification by Log Gabor filter optimization, Proceedings of Fourth International Conference on Advances in Computing and Communications, pp. 50-55, (2014)
[4]  
GIRSHICK R, DONAHUE J, DARRELL T, Et al., Rich feature hierarchies for accurate object detection and semantic segmentation, Proceedings of IEEE Conference on Computer Vision and Pattern Recognition, pp. 580-587, (2014)
[5]  
GU R, BILLINGE S J L, DU Q., A fast two-stage algorithm for non-negative matrix factorization in smoothly varying data, Acta Crystallographica Section A, Foundations and Advances, 79, 2, pp. 203-216, (2023)
[6]  
ZHAO W Q, XU M F, CHENG X F, Et al., An insulator in transmission lines recognition and fault detection model based on improved faster RCNN, IEEE Transactions on Instrumentation and Measurement, 70, pp. 1-8, (2021)
[7]  
LIU W, ANGUELOV D, ERHAN D, Et al., SSD: single shot multibox detector, European Conference on Computer Vision, pp. 21-37, (2016)
[8]  
REDMON J, DIVVALA S, GIRSHICK R, Et al., You only look once: unified, real-time object detection, Proceedings of IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 779-788, (2016)
[9]  
REDMON J, FARHADI A., YOLO9000: better, faster, stronger, Proceedings of IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 6517-6525, (2017)
[10]  
SHEN J Q, LIU N Z, SUN H, Et al., Vehicle detection in aerial images based on lightweight deep convolutional network and generative adversarial network, IEEE Access, 7, pp. 148119-148130, (2019)