Journal of Applied Optics, Volume. 45, Issue 4, 732(2024)

Small object detection algorithm based on improved YOLOv3

Kai WANG, Shuli LOU*, and Yan WANG
Author Affiliations
  • School of Physics and Electronic Information, Yantai University, Yantai 264005, China
  • show less

    In order to effectively solve the problem that small objects are difficult to recall and easy to miss detection, an improved YOLOv3 algorithm based on feature fusion and feature enhancement was proposed. To enhance the generalization performance of the model, the Mosaic and Mixup methods were combined for data enhancement during training. Firstly, in order to improve the low recall rate of small object detection, the original feature fusion network was extended to the shallower layer, and the bottom-up feature pyramid was added, so that the details and positioning information of the shallow feature layer could be transmitted to the deep layer. Secondly, a feature enhancement module was proposed to enlarge the receptive field, so that the shallow feature layer could obtain the rich deep semantic information and optimize the expression ability of the feature layer. Finally, the GIoU was used as a regression loss function to reduce the missing rate and achieve more accurate regression. Simulation experiments on Pascal VOC2007 and VOC2012 show that the improved algorithm can improve the mean average precision (mAP) value by 4.4% on the premise of ensuring the detection speed. Experimental results fully prove that the proposed algorithm can effectively improve the performance of small object detection.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Kai WANG, Shuli LOU, Yan WANG. Small object detection algorithm based on improved YOLOv3[J]. Journal of Applied Optics, 2024, 45(4): 732

    Download Citation

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

    Category: Research Articles

    Received: May. 31, 2023

    Accepted: --

    Published Online: Oct. 21, 2024

    The Author Email: LOU Shuli (娄树理)

    DOI:10.5768/JAO202445.0402002

    Topics