科研之友
微信
新浪微博
Facebook
分享链接
摘要
针对遥感图像目标检测算法漏检和误检率高、目标定位不精确、无法准确识别目标类别等问题，提出一种改进YOLOv8的目标检测算法。为提高模型的损失函数对梯度分配的灵活性，适应各种形状和尺寸的物体，设计了非单调聚焦机制与边界框几何因素相结合的边界框回归损失函数；为扩大模型的感受野并削弱遥感图像背景对检测目标的影响，采用全局注意力机制与残差块结合的方式，设计了残差全局注意力机制；为使模型适应遥感图像中目标物体的形变与不规则排列，对YOLOv8模型中的C2f模块进行改进，融入可变形卷积与可变形RoI池化层。实验结果表明，在DOTA数据集和RSOD数据集上，改进YOLOv8算法的mAP@0.5达到72.1%和94.6%，优于其它对比主流算法，提高了遥感图像目标检测的精度，为遥感图像识别提供了新的手段。
- 出版日期2023
- 单位南京信息工程大学
