Journal of Optoelectronics · Laser, Volume. 35, Issue 8, 793(2024)

Super-resolution and multi-scale fusion target detection algorithm based on improved YOLOv5

YAO Shanshan, WANG Jingyu, HAO Bin*, ZHANG Fei, GAO Lu, and REN Xiaoying
Author Affiliations
  • School of Information Engineering, Inner Mongolia University of Science and Technology, Baotou, Inner Mongolia 014000, China
  • show less

    To enhance the multi-scale learning capacity of target detection algorithms, particularly for small targets, this paper proposes a super-resolution and multi-scale fusion target detection algorithm based on an improved YOLOv5 framework. Firstly, instead of the up-sampling operation of the original YOLOv5 model, the algorithm utilizes sub-pixel convolution to enhance the image resolution and preserve the information of small targets to the greatest extent possible. Secondly, the algorithm utilizes the parallel fast multi-scale fusion (PFMF) module to achieve two-way fusion of deep and shallow features. This upgrade from the original YOLOv5 algorithm's 3-scale prediction to 4-scale prediction improves the model's ability to learn multi-scale features and detect small targets. The experimental results demonstrate that compared with YOLOv5s, the improved model achieves a 2.8% and 3.5% increase in mAP @0.5 and mAP @0.5∶0.95, respectively, on the PASCAL VOC dataset. Similarly, on the MS COCO dataset, the improved model achieves a 4.3% and 5.2% increase in mAP @0.5 and mAP @0.5∶0.95, respectively. The experiments demonstrate the improved YOLOv5 model's enhanced capability in multi-scale detection, particularly for small targets, and indicate its potential practical value.

    Tools

    Get Citation

    Copy Citation Text

    YAO Shanshan, WANG Jingyu, HAO Bin, ZHANG Fei, GAO Lu, REN Xiaoying. Super-resolution and multi-scale fusion target detection algorithm based on improved YOLOv5[J]. Journal of Optoelectronics · Laser, 2024, 35(8): 793

    Download Citation

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

    Category:

    Received: Dec. 19, 2022

    Accepted: Dec. 13, 2024

    Published Online: Dec. 13, 2024

    The Author Email: HAO Bin (sadream@imust.edu.cn)

    DOI:10.16136/j.joel.2024.08.0852

    Topics