• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2023, Vol. 45 ›› Issue (05): 878-884.

• Graphics and Images • Previous Articles     Next Articles

An improved YOLOv3 target detection optimization algorithm in dense traffic scenarios

HUO Ai-qing,ZHANG Shu-han,YANG Yu-yan,XU Jing-rong,WANG Ze-wen    

  1. (School of Electronic Engineering,Xi’an Shiyou University,Xi’an 710065,China)
  • Received:2021-10-25 Revised:2022-01-24 Accepted:2023-05-25 Online:2023-05-25 Published:2023-05-16

Abstract: Aiming at the problem of missed detection and false detection due to high overlap rate of detection targets in traffic congested scenes, a combination optimization algorithm (named L-YOLOv3+CIoU Loss+SD-NMS) containing improved YOLOv3, CIoU loss function optimization, and SD-NMS optimization is proposed. Deep separable convolution, SE module, and Ghost module are used to improve the residual unit structure of YOLOv3, in order to improve the ability of dense target feature extraction and reduce the amount of network model parameters. CIoU Loss is adopted to speed up the network  model convergence speed. Meanwhile, the multi-target set prediction idea is combined with DIoU-NMS to propose the SD-NMS optimization algorithm, in order to reduce the false detection rate of missed detection. Experimental results on the BDD100K data set show that the improved target detection algorithm has a recall rate of 91.58% and an accuracy rate of 93.04%. Compared with YOLOv3 algorithm, the proposal improves the recall rate and accuracy by 12.09% and 9.52%, respectively, showing better detection effect. 


Key words: target detection, deep learning, YOLOv3 algorithm, CIoU loss, non-maximum suppression