我们的服务符合学术规范和道德
专业 高端让您使用时没有后顾之忧

印刷与数字媒体技术研究2024年第05期:基于改进YOLOv7的SAR图像小目标检测方法(英文)

来源:期刊VIP网 时间:


作者:杨可;司占军;张滢雪;石金玉;

单位:天津科技大学人工智能学院;

摘要:针对当前合成孔径雷达(Synthetic Aperture Radar, SAR)图像目标检测方法不能够适应不同尺寸目标、图像背景复杂导致的检测精度低等问题,本研究提出了一种基于改进YOLOv7的SAR图像小目标检测方法。该方法对特征提取网络进行改进,在骨干网络中使用SAC卷积帮助模型捕获不同尺度的目标信息,提高对小目标的特征提取能力。同时基于注意力机制,在原始目标检测头上嵌入DyHead模块,在降低复杂背景影响的同时更好地筛选小目标。此外,引入NWD损失函数与CIoU损失函数相结合,相比于原CIoU损失函数,NWD损失函数更加注重对小目标的处理,从而进一步提高了对小目标的检测能力。通过在数据集HRSID上的实验结果表明,改进后方法的mAP@0.5、mAP@0.95分别为93.5%、71.5%,相比基线模型分别提升了7.2%、7.6%,其能够有效地完成SAR图像小目标检测任务。

关键词:小目标检测;;合成孔径雷达;;YOLOv7;;DyHead模块;;可切换的空洞卷积