Journal of System Simulation ›› 2022, Vol. 34 ›› Issue (6): 1219-1229.doi: 10.16182/j.issn1004731x.joss.21-0907

• Modeling Theory and Methodology • Previous Articles     Next Articles

Vehicle Detection Method Based on Multi Scale Feature Fusion

Yin Wang(), Feixiang Wang(), Qianlai Sun   

  1. College of Electronic Information Engineering, Taiyuan University of Science and Technology, Taiyuan 030024, China
  • Received:2021-09-03 Revised:2021-10-16 Online:2022-06-30 Published:2022-06-16
  • Contact: Feixiang Wang E-mail:xpw417@163.com;2601741160@qq.com

Abstract:

Vehicle detection is the important research content and hotspot in the intelligent transportation. Aiming at the low detection accuracy and poor small-scale recognition effect of the traditional vehicle detection algorithm, an improved detection method based on YOLOv4(you only look once v4) is proposed to improve the detection performance of small target vehicles in traffic scenes. By redesigning the YOLOv4 network, the MobileNetv2 deep separable convolution module is used to replace the traditional convolution, and the convolutional block attention module (CBAM) attention module is integrated into the feature extraction network to ensure the detection accuracy of the model and reduce the model parameters. The deep and shallow semantic information of the four scale feature maps is fused by using PANet-D feature fusion to enhance the detection ability of small objects. By using Focal loss to optimize the classification loss function, the convergence speed of the network model is accelerated. The experimental results show that the recognition accuracy of the improved network reaches 96.55%, and the size of the network model is 92.49 M lower, and the detection speed is 17% higher than those of the original YOLOv4 network, which fully proves the feasibility of the algorithm.

Key words: vehicle detection, YOLOv4, multiscale fusion, depth separable convolution, attention mechanism

CLC Number: