一种基于改进的YOLOv5的旋转目标检测方法

本发明公开了一种基于改进的YOLOv5的旋转目标检测方法。该发明在旋转目标检测方向上具有一定的通用性，该专利以遥感图像目标检测为说明案例。对于特征提取过程中提取到过多背景信息的问题，本文引入了注意力机制，并且对其做出改进，针对遥感图像中目标的特点，采用拥有更小感受野的卷积核来赋予模型空间上的注意力。残差模块采用C3结构构建更深层次的网络，SPP用于不同尺度特征的融合，并且引入角度分类的思想以及CSL标签，采用高斯函数作为窗口函数，让模型具有学习角度距离的能力。