Real-Time Object Detection Based on Improved YOLOv3 Network

被引:8
|
作者
Sun Jia [1 ]
Guo Dabo [1 ]
Yang Tiantian [1 ]
Ma Shitu [1 ]
机构
[1] Shanxi Univ, Coll Phys & Elect Engn, Taiyuan 030006, Shanxi, Peoples R China
关键词
machine vision; image processing; object detection; YOLOv3; k-means algorithm;
D O I
10.3788/LOP57.221505
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
For the shortcoming of the real-time performance of YOLOv3 algorithm in object detection, we propose an improved network structure and a new method for video object detection adapted to real-time object detection. Firstly, the proposed k-means-threshold (k-thresh) method makes up for the problem of its sensitivities to the initial position of the cluster center, and performs cluster analysis on a data set including three categories to select more appropriate anchor boxes. Then, the 1 X down-sampling and 8 X down-sampling feature maps arc stitched together into the third layer detection layer to improve the detection accuracy of the object and increase the the mean average precision of the YOLOv3 algorithm by 2%. Finally, the camera captures the image and the excellent detection data obtained in the previous period to predict the target of the new image and adds a re-detection threshold to improve the smoothness of video detection. The experimental results show that the proposed improved YOLOv3 network improves the detection accuracy and the real-time performance, the maximum frame rate reaches 61.26 frame/s in 30 min of real-time detection, which is 1 times faster than the original YOLOv3 algorithm.
引用
收藏
页数:10
相关论文
共 29 条
  • [1] [Anonymous], 2010, Proceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition, DOI DOI 10.1109/CVPR.2010.5539906
  • [2] A comparative study of efficient initialization methods for the k-means clustering algorithm
    Celebi, M. Emre
    Kingravi, Hassan A.
    Vela, Patricio A.
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2013, 40 (01) : 200 - 210
  • [3] 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
  • [4] Detect to Track and Track to Detect
    Feichtenhofer, Christoph
    Pinz, Axel
    Zisserman, Andrew
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, : 3057 - 3065
  • [5] Felzenszwalb P, 2008, PROC CVPR IEEE, P1984
  • [6] Aerial Target Detection Based on Improved Faster R-CNN
    Feng Xiaoyu
    Mei Wei
    Hu Dashuai
    [J]. ACTA OPTICA SINICA, 2018, 38 (06)
  • [7] Object Detection with Discriminatively Trained Part-Based Models
    Forsyth, David
    [J]. COMPUTER, 2014, 47 (02) : 6 - 7
  • [8] Rich feature hierarchies for accurate object detection and semantic segmentation
    Girshick, Ross
    Donahue, Jeff
    Darrell, Trevor
    Malik, Jitendra
    [J]. 2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, : 580 - 587
  • [9] Joseph RK, 2016, CRIT POL ECON S ASIA, P1
  • [10] 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)