Improved remote sensing image target detection based on yolov7
DOI:
Author:
Affiliation:

school

Clc Number:

Fund Project:

This work is supported by the National Natural Science Foundation of China under Grant No. 62005196 and 61974104.

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Remote sensing images are taken at high altitude from above, with complex spatial scenes of images and a large number of target types. The detection of image targets on large scale remote sensing images suffers from the problem of small target size and target density.The paper here proposes an improved model for remote sensing image detection based on YOLOv7. First, the small-scale detection layer is added to reacquire tracking frames to improve the network's recognition ability of small-scale targets; then Bottleneck Transformers are fused in the backbone to make full use of the CNN+Transformer architecture to enhance the feature extraction ability of the network; after that, the CBAM attention mechanism is added in the head to improve the model's ability of small-scale target; Finally, the non-maximum suppressed NMS of YOLOv7 algorithm is changed to DIOU_NMS to improve the detection ability of overlapping targets in the network. The results show that the method in this paper can improve the detection rate of small-scale targets in remote sensing images and effectively solve the problem of high overlap, and is tested on the NWPU-VHR10 and DOTA datasets, and the accuracy of the improved model is improved by 6.3% and 4.2%, respectively, compared with the standard yolov7 algorithm.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:April 08,2023
  • Revised:September 21,2023
  • Adopted:October 08,2023
  • Online:
  • Published: