[发明专利]一种基于Efficientdet的旋转目标检测方法、设备及存储设备

权利要求书

1.一种基于Efficientdet的旋转目标检测方法，其特征在于：包括以下步骤：

S101：构建具有双向跨尺度连接特征融合结构的遥感图像旋转目标检测网络结构；设计旋损失的八参数损失函数，得到最终的角度位置损失函数，将该最终的角度位置损失函数与所述遥感图像旋转目标检测网络结构进行结合，得到最终的网络结构；

S102：采用迁移学习对所述遥感图像旋转目标检测网络结构进行训练，得到训练完成的网络结构；

最终的网络结构训练过程中，采用改进的EfficientNet网络对输入图像进行各层特征提取，并利用双向跨尺度连接特征融合网络对提取的各层特征进行融合，得到若干输出特征层；

将所述输出特征层输入至检测层，即得到训练完成的网络；

S103：将遥感数据集输入至所述训练完成的网络结构，得到遥感图像的目标检测结果。

2.如权利要求1所述的一种基于Efficientdet的旋转目标检测方法，其特征在于：步骤S101中，所述八参数为边界框中四个角的点坐标，由以下几个部分取得，i)将预测框的四个顶点顺时针移动一个位置；ii)保持预测框顶点的顺序不变；iii)将预测框的四个顶点逆时针移动一个位置；iv)取上述三种情况下的最小值。

3.如权利要求1所述的一种基于Efficientdet的旋转目标检测方法，其特征在于：步骤S101中，所述八参数损失函数为：

其中，x_i和y_i分别表示预测框的第i个顶点和参考框的第i个顶点之间的坐标偏移，和分别表示地面真值框的第i个顶点与参考框的第i个顶点之间的偏移，x_(i+3)％4与y_(i+3)％4代表将预测框的四个顶点顺时针移动一个位置后的坐标，x_i与y_i代表保持预测框顶点的顺序不变的坐标，x_(i+1)％4与y_(i+1)％4代表将预测框的四个顶点逆时针移动一个位置的坐标。

4.如权利要求1所述的一种基于Efficientdet的旋转目标检测方法，其特征在于：步骤S101中，所述最终的角度位置损失函数为：

其中，N表示基于Efficientdet的旋转目标检测anchor的个数，t'_n取值为0或1，v'_nj表示预测的偏移矢量，v_nj表示真值框的目标向量，t_n为目标类别，p_n为sigmoid计算的各类别概率分布，L_cls为focal loss，L_reg为smooth L1 loss，λ₁、λ₂均表示超参数。

5.如权利要求1所述的一种基于Efficientdet的旋转目标检测方法，其特征在于：步骤S102中，所述改进EfficientDet网络是指，将原来的水平框改为旋转框。

6.一种存储设备，其特征在于：所述存储设备存储指令及数据用于实现权利要求1～5任一项所述的一种基于Efficientdet的旋转目标检测方法。

7.一种基于Efficientdet的旋转目标检测设备，其特征在于：包括：处理器及存储设备；所述处理器加载并执行所述存储设备中的指令及数据用于实现权利要求1～5任一项所述的一种基于Efficientdet的旋转目标检测方法。