基于YOLOv7的光学遥感图像目标检测

Optical Remote Sensing Image Object Detection Based on YOLOv7

  • 摘要: 针对光学遥感图像目标检测中目标尺度差异大和背景复杂造成的检测困难问题,本文提出一种基于YOLOv7的光学遥感图像目标检测算法,该算法分别对特征提取和特征融合过程进行了优化。首先,引入CNN和Transformer相结合的特征提取模块以更好地捕捉图像的全局信息;然后,设计双向融合结构增强浅层特征与深层特征的融合效果。实验结果表明,该方法的mAP@0.5在NWPU VHR-10数据集和RSOD数据集上分别达到96.6%和97.6%,较YOLOv7算法提升了3.2%和4.2%,有效提高了遥感图像目标检测的精度。

     

    Abstract: To address the challenge of significant differences in target scales and complex backgrounds in optical remote sensing image target detection, this paper presents an optical remote sensing image target detection algorithm based on YOLOv7. The algorithm optimizes the feature extraction and fusion processes. First, a feature extraction module combining a CNN and Transformer is introduced to better capture the global information of the image. Subsequently, a bidirectional fusion structure is designed to enhance the fusion effect between the shallow and deep features. The experimental results demonstrate that the proposed method achieves mAP@0.5 of 96.6% and 97.6% on the NWPU VHR-10 and RSOD datasets, respectively, surpassing that achieved by the YOLOv7 algorithm by 3.2% and 4.2%, respectively; thus, effectively enhancing the accuracy of remote sensing image target detection.

     

/

返回文章
返回