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

Computer Engineering & Science

Previous Articles     Next Articles

A training set optimization and detection
method based on YOLOv3 algorithm

GAO Xing1,LIU Jian-fei1,HAO Lu-guo2,DONG Qi-qi1   

  1.  (1.School of Electronic Information Engineering,Hebei University of Technology,Tianjin 300401;
    2.School of Information Engineering,Guangdong University of Technology,Guangzhou 510006,China)
     
  • Received:2019-05-27 Revised:2019-08-05 Online:2020-01-25 Published:2020-01-25

Abstract:

YOLOv3 is a one-stage target detection algorithm that does not generate Region Proposal Network (RPN) to extract target information. Compared with the two-stage target detection algorithm, it has faster detection speed. However, the existing algorithms has the problems of low accuracy and missed detection in small target detection. Therefore, based on YOLOv3 algorithm, a detection method of training set optimization and layer processing is proposed. Firstly, K-means algorithm is used to cluster the standard dataset VOC2007+2012 and the self-built behavior dataset, so as to get the anchor size that fits the training size of the dataset. Then, The training is carried out by adjusting the training parameters and selecting a reasonable labeling method. Finally, the input image is processed by layer and the target is detected. The experimental results show that the average accuracy (mAP) of VOC2007 verification set is improved by 1.4% after clustering analysis, the problem of small feeling field on the higher convolution layer is effectively solved during the detection process of the original algorithm, and the accuracy of YOLOv3 algorithm is improved when detecting small target objects and reducing the missed detection rate.
 

Key words: YOLOv3, anchor, small target detection, cluster analysis