(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210220243.0 (22)申请日 2022.03.08 (71)申请人 上海交通大 学 地址 200240 上海市闵行区东川路80 0号 (72)发明人 王若谷　李柠　 (74)专利代理 机构 上海光华专利事务所(普通 合伙) 31219 专利代理师 庞红芳 (51)Int.Cl. G06V 10/22(2022.01) G06V 10/24(2022.01) G06V 10/44(2022.01) G06V 10/762(2022.01) G06V 10/82(2022.01) G06K 9/62(2022.01)G06N 3/04(2006.01) G06N 3/08(2006.01) (54)发明名称 旋转目标检测方法、 存 储介质及电子设备 (57)摘要 本发明提供一种旋转目标检测方法、 存储介 质及电子设备， 所述方法包括： 基于旋转目标检 测的数据集确定锚框； 通过第一神经网络和所述 锚框生成待检测物体的检测框， 并预测所述检测 框的四个顶 点坐标； 通过第二神经网络预测所述 检测框的中心坐标和类别； 对所述检测框和所述 检测框的四个顶点坐标进行矩形约束修正， 生成 第一修正矩形框； 基于预测的所述中心坐标对所 述第一修正矩形框的位置进行修正， 生成第二修 正矩形框； 根据所述第二修正矩形框的坐标输出 所述待检测物体的旋转角本发明可以有效解决 现有技术中难以准确预测旋转目标方向角的技 术问题， 有效解决旋转目标检测问题。 权利要求书2页 说明书8页 附图5页 CN 114677498 A 2022.06.28 CN 114677498 A 1.一种旋转目标检测方法， 其特 征在于： 所述方法包括： 基于旋转目标检测的数据集确定锚框； 通过第一神经网络和所述锚框生成待检测物体的检测框， 并预测所述检测框的四个顶 点坐标； 通过第二神经网络预测所述检测框的中心坐标和类别； 对所述检测框和所述检测框的四个顶点坐标进行矩形约束修正， 生成第一修正矩形 框； 基于预测的所述中心坐标对所述第 一修正矩形框的位置进行修正， 生成第 二修正矩形 框； 根据所述第二 修正矩形框的坐标输出 所述待检测物体的旋转角。 2.根据权利 要求1所述的旋转目标检测方法， 其特征在于： 采用kmeans++聚类算法确定 锚框。 3.根据权利要求1或2所述的旋转目标检测方法， 其特征在于： 所述基于旋转目标检测 的数据集确定锚框包括： 设置聚类中心数目； 从数据集中随机 选取一个样本作为初始聚类中心； 计算数据集中每个样本与当前聚类中心之间的最短距离， 并计算每个样本被选作 下一 个聚类中心的概 率， 重复本步骤， 获得个数与所述聚类中心数目对应的聚类中心； 计算每个样本到各个聚类中心的距离， 将其分类到距离最小的聚类中心所对应的类别 中， 并针对每个类别， 计算当前属于该类别的样本的中位数作为新的聚类中心， 重复本步 骤， 直到每 个样本的分类不在变化， 此时得到的聚类中心即为锚框 。 4.根据权利要求3所述的旋转目标检测方法， 其特征在于： 采用数据集标签真实框和所 选聚类中心的边界框的交并比计算数据集中每 个样本与当前聚类中心之间的最短距离 。 5.根据权利要求1所述的旋转目标检测方法， 其特征在于： 所述第 一神经网络的神经网 络模型为Dark Net53； 所述第二神经网络的神经网络模型为CS PDarkNet53。 6.根据权利要求1所述的旋转目标检测方法， 其特征在于： 所述对所述检测框和所述检 测框的四个顶点 坐标进行矩形约束修 正， 生成第一 修正矩形框包括： 获取预测的所述检测框的四个顶点 坐标； 根据预测的所述检测框的四个顶点坐标的位置关系计算包络所述检测框的四个顶点 的最小外 接矩形， 即为第一 修正矩形框； 输出所述最小外接矩形四个角点的坐标。 7.根据权利要求6所述的旋转目标检测方法， 其特征在于： 所述基于预测的所述中心坐 标对所述第一 修正矩形框的位置进行修 正， 生成第二 修正矩形框包括： 将所述第一 修正矩形框的坐标按左上、 右上、 右下、 左下的顺序排列； 选取左上、 右下或右上、 左下两个点的坐标计算 其之间的中心坐标； 利用所述预测的中心坐标对所述第 一修正矩形框的中心坐标进行修正， 同时根据两者 间的偏差对所述第一修正矩形框的四个顶点坐标进行修正， 得到最 终第二修正矩形框的中 心坐标和四个顶点 坐标。 8.根据权利要求7所述的旋转目标检测方法， 其特征在于： 选取所述第 二修正矩形框的权　利　要　求　书 1/2 页 2 CN 114677498 A 2左上、 右上或左下、 右下两个点的坐标计算所述待检测物体的旋转角。 9.一种存储介质， 存储有程序指令， 其特征在于： 所述程序指令被执行时实现如权利要 求1至权利要求8任一项所述的旋转目标检测方法的步骤。 10.一种电子设备， 其特征在于： 包括存储器， 用于存储计算机程序； 处理器， 用于运行 所述计算机程序以实现如权利要求1至权利要求8任一权利要求所述的旋转 目标检测方法 的步骤。权　利　要　求　书 2/2 页 3 CN 114677498 A 3