基于改进YOLOv7的遥感图像舰船目标检测算法

    Remote sensing image ship target detection algorithm based on improved YOLOv7

    • 摘要: 针对遥感图像中舰船目标存在尺度变化大、大长宽比、排列密集、背景信息复杂等特点,提出了一种基于改进YOLOv7的遥感图像舰船目标检测算法。以YOLOv7作为基线网络,优化数据集先验锚框生成算法;使用长边表示法结合圆形平滑标签方法,解决边界回归的周期性带来的突变问题;并将全局注意力(GAM)与无参注意力(SimAM)嵌入到YOLOv7网络中,有效消弱遥感图像中复杂背景区域信息带来的干扰,提高目标检测准确率;优化坐标框损失函数,加快模型收敛速度。通过对舰船数据集(DOTA-ship),以及HRSC2016数据集执行单类检测与多类检测任务,m AP结果分别为86.1%、97.7%、87.1%,相比于原YOLOv7分别提升了7.8%、4.6%、7.9%,验证了方法的有效性与优越性。

       

      Abstract: To address challenges such as significant scale variations, high aspect ratios, dense arrangements, and complex backgrounds in ship target detection from remote sensing images, this paper proposes an improved YOLOv7-based algorithm. Using YOLOv7 as the baseline network, the prior anchor generation algorithm is optimized for the dataset. A long-edge representation method combined with circular smooth labeling is introduced to detect ship targets with uncertain rotation directions. The YOLOv7 network is enhanced by embedding both the GAM(Global Attention Mechanism) and Sim AM(Simple Attention Mechanism) modules, which effectively suppress interference from complex background regions in remote sensing images and improve target detection accuracy. Additionally, the coordinate loss function is optimized to accelerate model convergence. Experimental results on the DOTA-ship and HRSC2016 datasets for both single-class and multi-class detection tasks show m AP values of 86. 1%, 97. 7%, and 87. 1%, respectively-representing improvements of 7. 8%, 4. 6%, and 7. 9% over the original YOLOv7 model. These results validate the effectiveness and superiority of the proposed method.

       

    /

    返回文章
    返回