Electronics Optics & Control, Volume. 31, Issue 4, 28(2024)

Multi-scale Remote Sensing Small Target Detection Based on cosSTR-YOLOv7

ZHANG Xu1... ZHU Zhengwei1,2, GUO Yuying1,2, LIU Hui1 and ZHONG Hui1 |Show fewer author(s)
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less

    In geospatial remote sensing images,the target detection accuracy is low due to dense target distribution,large target scale variation range and too little feature information of small targets.To solve the problem,a multi-scale remote sensing small target detection algorithm cosSTR-YOLOv7 based on Swin Transformer (STR) and YOLOv7 is proposed.YOLOv7 is taken as the baseline network.Firstly,STR module is used to replace E-ELAN module in the backbone network,and it is improved to be cosSTR module by using cosine attention mechanism and post-regularization method,so as to improve the stability of model training.Secondly,a new feature fusion layer is constructed in the Neck part to reduce feature information loss.A small target prediction layer is added to the prediction part to improve the model’s ability of small target detection.Finally,a new SIoU loss function is used to calculate the positioning loss to accelerate the convergence rate of the model.The remote sensing dataset DIOR is used for the experiment,and the experimental results show that the mean Average Precision (mAP) of the proposed algorithm reaches 92.63%which is 3.73 percentage points higher than that of the original YOLOv7 algorithmand the performance of multi-scale small target detection has been significantly improved.

    Tools

    Get Citation

    Copy Citation Text

    ZHANG Xu, ZHU Zhengwei, GUO Yuying, LIU Hui, ZHONG Hui. Multi-scale Remote Sensing Small Target Detection Based on cosSTR-YOLOv7[J]. Electronics Optics & Control, 2024, 31(4): 28

    Download Citation

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

    Category:

    Received: Mar. 12, 2023

    Accepted: --

    Published Online: Jul. 30, 2024

    The Author Email:

    DOI:10.3969/j.issn.1671-637x.2024.04.006

    Topics