Journal of Jilin University Science Edition ›› 2024, Vol. 62 ›› Issue (3): 665-673.

Previous Articles     Next Articles

Target Recognition Algorithm of Traffic Intersection Based on Improved YOLOv7

JIANG Sheng1, ZHANG Zhongyi1,2, WANG Zongyang2, YU Qing1   

  1. 1. School of Physics, Changchun University of Science and Technology, Changchun 130022, China;
    2. Institute of Deep Perception Technology, Wuxi 214000, Jiangsu Province, China
  • Received:2023-06-12 Online:2024-05-26 Published:2024-05-26

Abstract: Aiming at the problems of low accuracy, under-detection, and missed detection in the vehicle target detection algorithm at traffic intersections, we proposed a target recognition algorithm of traffic intersection based on improved YOLOv7.  Firstly, the algorithm  used the feed-forward convolutional attention mechanism CBAM to enhance the network’s  attention to key features from both channel attention and spatial attention, improve the network’s running  speed, and optimize the network’s feature extraction capabilities. Secondly, a new learning module was formed by connecting the  spatial layer to depth  layers to form a  full-dimensional dynamic convolution, which improved the YOLOv7 feature learning method and enhanced the feature expression ability. Finally, the experiments were conducted on the actual collected traffic intersection dataset. The experimental results show that the proposed method  achieves an average accuracy of 96.1% on the corresponding dataset, and the training time is reduced to 16.71 h. Therefore, it has obvious recognition advantages  for small target detection at traffic intersections.

Key words: deep learning, target detection, convolutional neural network, attention mechanism, full-dimensional dynamic convolution

CLC Number: 

  • TP301.6