一种基于YOLOv8的卫星遥感旋转目标检测方法及系统

本发明公开了一种基于YOLOv8的卫星遥感旋转目标检测方法及系统，涉及卫星遥感目标检测技术领域。针对传统一阶段目标检测算法在遥感图像上检测精度低和速度慢的问题。首先，对YOLO主干特征提取网络的最后一层下采样模块和检测层的大目标检测头进行了裁剪，并增加了一个小目标检测头，使模型更好地关注遥感小目标。然后，设计了一个快速特征提取模块嵌入到主干特征提取网络中以实现高效的特征提取。此外，结合EMA注意力机制设计出一个高性能特征融合模块，嵌入到主干网络和特征融合网络之间实现模型对关键特征信息表征能力的增强。最后，通过将目标边界框转化为高斯分布，通过r‑KLDloss损失函数计算损失值，帮助模型更好地学习特征信息以提高检测精度。