Improved Real-Time Vehicle Detection Method Based on YOLOV3

被引:6
|
作者
Li Hanbing [1 ]
Xu Chunyang [1 ]
Hu Chaochao [1 ]
机构
[1] Jiangsu Univ, Sch Automot & Traff Engn, Zhenjiang 212013, Jiangsu, Peoples R China
关键词
machine vision; vehicle; target detection; YOLOV3; inverted residual network; real-time detection;
D O I
10.3788/LOP57.101507
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Aiming at the problem of low real-time performance of original YOLOV3 target detection algorithm in vehicle detection tasks, this paper proposes an improved vehicle detection model. In order to reduce the number of parameters, reduce the computational complexity, and solve the problem of gradient disappearance and gradient explosion, the model uses the inverted residual network as the basic feature extraction layer. In addition, group normalization is used to reduce the impact of batch size on the accuracy of the model. At the same time, softening non maximum suppression is used to reduce the rate of missed detection. Finally, the Focal-loss is used to improve the loss function so that the model focuses on the difficult-to-classify samples in the process of training. The parameter amount of the improved model is 36.23% of YOLOV3 model. The detection time per frame is reduced by 13.8 ms compared with YOLOV3, and the average category accuracy is improved by 1.15%. The results show that the proposed algorithm ensures both real-time performance and accuracy, and providing a reference for real-time detection of vehicles.
引用
收藏
页数:7
相关论文
共 21 条
  • [1] Speeded-Up Robust Features (SURF)
    Bay, Herbert
    Ess, Andreas
    Tuytelaars, Tinne
    Van Gool, Luc
    [J]. COMPUTER VISION AND IMAGE UNDERSTANDING, 2008, 110 (03) : 346 - 359
  • [2] Bodla N, 2017, 2017 IEEE INT C COMP
  • [3] Cheng Y H, 2019, COMPUTER SYSTEMS APP, V28, P107
  • [4] CHOLLET F, 2017, PROC CVPR IEEE, P1800, DOI [DOI 10.1109/CVPR.2017.195, 10.1109/CVPR.2017.195]
  • [5] Histograms of oriented gradients for human detection
    Dalal, N
    Triggs, B
    [J]. 2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 1, PROCEEDINGS, 2005, : 886 - 893
  • [6] Girshick R, 2011, PROC OF IEEE CONF ON, P580
  • [7] Contextual Action Recognition with R*CNN
    Gkioxari, Georgia
    Girshick, Ross
    Malik, Jitendra
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1080 - 1088
  • [8] He K., 2016, Proceedings of the IEEE conference on computer vision and pattern recognition, DOI DOI 10.1109/CVPR.2016.90
  • [9] Improved YOLO V3 Algorithm and Its Application in Small Target Detection
    Ju Moran
    Luo Haibo
    Wang Zhongbo
    He Miao
    Chang Zheng
    Hui Bin
    [J]. ACTA OPTICA SINICA, 2019, 39 (07)
  • [10] [李浩 Li Hao], 2017, [交通信息与安全, Journal of Transport Information and Safety], V35, P37