[发明专利]基于辅助框回归的旋转目标检测方法、装置和存储介质

说明书

本申请提供一种基于辅助框回归的旋转目标检测方法、装置和存储介质，应用于Anchor‑free网络检测模型。该方法包括：获取目标物体的图片；将目标物体的图片发送给预先构建的旋转目标检测网络模型进行预测，并获得至少两个中间预测结果；基于目标物体的初始正向包裹框和预测正向包裹框，对至少两个中间预测结果进行过滤，以获得最终预测结果；其中初始正向包裹框与目标物体的初始旋转目标框对应，预测正向包裹框与目标物体的预测旋转目标框对应。在推理阶段，通过根据目标物体的初始正向包裹框和预测正向包裹框，中间预测结果进行优化过滤，能够获得更准确的预测正向包裹框信息，从而使得Anchor‑free模型在保证速度优势的同时，还能有效地提升精度。

技术领域

本申请涉及计算机视觉技术领域，更具体地涉及一种基于辅助框回归的旋转目标检测方法、装置和存储介质。

背景技术

在目标检测中，常用的水平矩形框很难贴合某些图像中具有多种角度的目标，往往需要用旋转矩形框来代替。如何对旋转矩形框进行有效的表示以及回归，这在一定程度上关系到目标检测模型的训练难度和预测精度。现阶段，面向旋转目标检测的模型主要有Anchor-based模型和Anchor-free模型，但Anchor-based模型精度更高，但运算量较大，Anchor-free模型速度较快但损失了精度。因此需要改进。

发明内容

为了解决上述问题中而提出了本申请。根据本申请一方面，提供了一种基于辅助框回归的旋转目标检测方法，应用于Anchor-free网络检测模型，所述方法包括：

获取目标物体的图片；

将所述目标物体的所述图片发送给预先构建的旋转目标检测网络模型进行预测，并获得至少两个中间预测结果；

基于所述目标物体的初始正向包裹框和预测正向包裹框，对所述至少两个中间预测结果进行过滤，以获得最终预测结果；

其中所述初始正向包裹框与所述目标物体的初始旋转目标框对应，所述预测正向包裹框与所述目标物体的预测旋转目标框对应。

在本申请的一个实施例中，将所述目标物体的图片发送给预先构建的旋转目标检测网络模型进行预测之前，所述方法还包括：

构建所述旋转目标检测网络模型。

在本申请的一个实施例中，构建所述旋转目标检测网络模型，包括：

获取已标注所述初始旋转目标框的样本集；

根据所述始始旋转目标框确定与所述初始旋转目标框对应的所述初始正向包裹框，并计算与所述初始正向包裹框对应的所述预测正向包裹框；

计算所述初始正向包裹框对比真实正向包裹框之间的第一损失值，并计算所述预测正向包裹框对比真实正向包裹框之间的第二损失值；

根据所述第一损失值和所述第二损失值确定所述旋转目标检测网络模型。

在本申请的一个实施例中，基于所述目标物体的初始正向包裹框和预测正向包裹框，对所述至少两个中间预测结果进行过滤，以获得最终预测结果，包括；

基于所述目标物体的初始正向包裹框和预测正向包裹框，采用非极大值抑制法对所述中间预测结果进行过滤，以获得最终预测结果。

在本申请的一个实施例中，基于所述目标物体的初始正向包裹框和预测正向包裹框，采用非极大值抑制法对所述中间预测结果进行过滤，以获得最终预测结果，包括：

计算所述初始正向包裹框和所述预测正向包裹框的回归得分；

将所述旋转目标检测网络模型的分类得分与所述回归得分进行加权计算，以得到置信度值。

在本申请的一个实施例中，计算所述初始正向包裹框和所述预测正向包裹框的回归得分，包括：