The design of lightweight vehicle detection model based on improved YOLOv5

Abstract

Convolutional neural network-based target detection algorithms are widely used in vehicle detection due to their high speed and accuracy. However, existing algorithms are characterized by large computational volumes, complex network structures, and severe resource constraints. They make them difficult to be ported to mobile platforms and embedded devices. Therefore, the structure of the relevant target detection algorithm needs to be optimized to enable wider deployment of the algorithm. To address the problems mentioned earlier, a YOLOv5SCB lightweight target detection network model is proposed. In the presented model, Shufflenetv2 and CA module are introduced into the backbone network to reduce the complexity of the network model and improve the detection accuracy of the model. Furthermore, BiFPN is integrated into the neck network to improve the efficiency of network feature fusion and enhance the ability of network feature expression. The experimental data show that compared with the original YOLOv5, the model parameters of the proposed YOLOv5SCB are reduced by 62.4% and the overall detection accuracy is improved by 1.1%.