Acta Optica Sinica, Volume. 39, Issue 7, 0715004(2019)

Improved YOLO V3 Algorithm and Its Application in Small Target Detection

Moran Ju1,2,3,4,5、*, Haibo Luo1,2,4,5, Zhongbo Wang1,2,3,4,5, Miao He1,2,3,4,5, Zheng Chang1,2,4,5, and Bin Hui1,2,4,5
Author Affiliations
  • 1 Shenyang Institute of Automation, Chinese Academy of Sciences, Shenyang, Liaoning 110016, China
  • 2 Institutes for Robotics and Intelligent Manufacturing, Chinese Academy of Sciences, Shenyang, Liaoning 110016, China
  • 3 University of Chinese Academy of Sciences, Beijing 100049, China
  • 4 Key Laboratory of Opt-Electronic Information Processing, Chinese Academy of Sciences, Shenyang, Liaoning 110016, China
  • 5 The Key Laboratory of Image Understanding and Computer Vision, Shenyang, Liaoning 110016, China
  • show less

    This study proposes an improved detection algorithm of YOLO V3 specially applied in small target detection to solve the problems of low detection and high false alarm rates of small targets in an image. The resolution of small targets is low, and their features are not obvious; thus, this study proposes 2× upsampling for the feature map down-sampled by 8× of the previous network,and the feature map upsampled by 2× is concatenated with the output of the second ResNet block unit. A feature fusion target detection layer, whose feature map is down-sampled by 4×, is established. Two ResNet units in the second ResNet block unit of Darknet53 in the YOLO V3 network structure are added to obtain more features of the small target. The K-means clustering algorithm is used to select the number of candidate anchor boxes and aspect ratio dimensions. A comparative experiment is performed based on the improved YOLO V3 algorithm on the VEDAI dataset and YOLO V3 algorithm. The results show that the improved YOLO V3 algorithm can efficiently detect small targets and improve the mean average precision and recall rate of small targets.

    Tools

    Get Citation

    Copy Citation Text

    Moran Ju, Haibo Luo, Zhongbo Wang, Miao He, Zheng Chang, Bin Hui. Improved YOLO V3 Algorithm and Its Application in Small Target Detection[J]. Acta Optica Sinica, 2019, 39(7): 0715004

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Machine Vision

    Received: Jan. 11, 2019

    Accepted: Mar. 22, 2019

    Published Online: Jul. 16, 2019

    The Author Email: Ju Moran (jumoran@sia.cn)

    DOI:10.3788/AOS201939.0715004

    Topics