检测模型训练方法以及目标检测方法技术

本说明书实施例提供检测模型训练方法以及目标检测方法，其中所述检测模型训练方法包括：获取标注有目标框的样本图像；将所述样本图像输入预设检测模型中进行目标检测，获得预测框；根据所述目标框的边信息和点信息，以及所述预测框的边信息和点信息，计算损失值；基于所述损失值调整所述预设检测模型的模型参数，返回执行所述获取标注有目标框的样本图像的步骤，在达到预设训练停止条件的情况下，确定训练后的所述预设检测模型为目标检测模型。通过边信息和点信息确定损失值，提高了模型检测的精确度。本方法实现简单，适用于使用任何深度学习框架，避免了方形物体无法回归的问题。题。题。

【技术实现步骤摘要】
检测模型训练方法以及目标检测方法


[0001]本说明书实施例涉及计算机
，特别涉及一种检测模型训练方法。

技术介绍

[0002]旋转目标检测是计算机视觉中的一项基本任务，相比于通用的目标检测而言，其检测的精度更高，在遥感、文字检测等场景中有着广泛的用途。旋转目标检测通常有两种做法，一种是用一个旋转矩形框去表示目标的位置，另一种是用一个多边形框去表示目标的位置。不管用哪种方法，旋转目标检测领域，目标检测的效率低。因此，亟需一种有效的方案以解决上述问题。

技术实现思路

[0003]有鉴于此，本说明书施例提供了一种检测模型训练方法。本说明书一个或者多个实施例同时涉及一种目标检测方法，一种检测模型训练装置，一种目标检测装置，一种计算设备，一种计算机可读存储介质以及一种计算机程序，以解决现有技术中存在的技术缺陷。
[0004]根据本说明书实施例的第一方面，提供了一种检测模型训练方法，包括：
[0005]获取标注有目标框的样本图像；
[0006]将所述样本图像输入预设检测模型中进行目标检测，获得预测框；
[0007]根据所述目标框的边信息和点信息，以及所述预测框的边信息和点信息，计算损失值；
[0008]基于所述损失值调整所述预设检测模型的模型参数，返回执行所述获取标注有目标框的样本图像的步骤，在达到预设训练停止条件的情况下，确定训练后的所述预设检测模型为目标检测模型。
[0009]根据本说明书实施例的第二方面，提供了一种目标检测方法，包括：
[0010]获取待检测图像；
[0011]将所述待检测图像输入训练好的目标检测模型，得到所述待检测图像的目标检测结果，其中，所述目标检测模型基于上述目标检测模型训练方法训练得到。
[0012]根据本说明书实施例的第三方面，提供了一种检测模型训练装置，包括：
[0013]第一获取模块，被配置为获取标注有目标框的样本图像；
[0014]检测模块，被配置为将所述样本图像输入预设检测模型中进行目标检测，获得预测框；
[0015]计算模块，被配置为根据所述目标框的边信息和点信息，以及所述预测框的边信息和点信息，计算损失值；
[0016]调整模块，被配置为基于所述损失值调整所述预设检测模型的模型参数，返回执行所述获取标注有目标框的样本图像的步骤，在达到预设训练停止条件的情况下，确定训练后的所述预设检测模型为目标检测模型。
[0017]根据本说明书实施例的第四方面，提供了一种目标检测装置，包括：
[0018]第二获取模块，被配置为获取待检测图像；
[0019]输入模块，被配置为将所述待检测图像输入训练好的目标检测模型，得到所述待检测图像的目标检测结果，其中，所述目标检测模型基于上述目标检测模型训练方法训练得到。
[0020]根据本说明书实施例的第五方面，提供了一种计算设备，包括：
[0021]存储器和处理器；
[0022]所述存储器用于存储计算机可执行指令，所述处理器用于执行所述计算机可执行指令，该计算机可执行指令被处理器执行时实现上述检测模型训练方法或目标检测方法的步骤。
[0023]根据本说明书实施例的第六方面，提供了一种计算机可读存储介质，其存储有计算机可执行指令，该指令被处理器执行时实现上述检测模型训练方法或目标检测方法的步骤。
[0024]根据本说明书实施例的第七方面，提供了一种计算机程序，其中，当所述计算机程序在计算机中执行时，令计算机执行上述检测模型训练方法或目标检测方法的步骤。
[0025]本说明书提供的一种检测模型训练方法，获取标注有目标框的样本图像；将所述样本图像输入预设检测模型中进行目标检测，获得预测框；根据所述目标框的边信息和点信息，以及所述预测框的边信息和点信息，计算损失值；基于所述损失值调整所述预设检测模型的模型参数，返回执行所述获取标注有目标框的样本图像的步骤，在达到预设训练停止条件的情况下，确定训练后的所述预设检测模型为目标检测模型。通过边信息和点信息确定损失值，而避免采用角度变量的直接距离度量，将角度差异统一转化成了边信息和点信息的差异，从而避免了角度周期性导致的目标不准确的问题，提高了模型检测的精确度。本方法实现简单，适用于使用任何深度学习框架，避免了方形物体无法回归的问题。
附图说明
[0026]图1是本说明书一个实施例提供的一种检测模型训练系统下检测模型训练的处理流程图；
[0027]图2是本说明书一个实施例提供的另一种检测模型训练系统的框架图；
[0028]图3是本说明书一个实施例提供的一种检测模型训练方法的流程图；
[0029]图4是本说明书一个实施例提供的另一种检测模型训练方法的流程图
[0030]图5是根据本说明书一个实施例提供的一种目标检测方法的流程图；
[0031]图6是本说明书一个实施例提供的一种目标检测方法的处理过程流程图；
[0032]图7是本说明书一个实施例提供的一种检测模型训练装置的结构示意图；
[0033]图8是本说明书一个实施例提供的一种目标检测装置的结构示意图；
[0034]图9是本说明书一个实施例提供的一种计算设备的结构框图。
具体实施方式
[0035]在下面的描述中阐述了很多具体细节以便于充分理解本说明书。但是本说明书能够以很多不同于在此描述的其它方式来实施，本领域技术人员可以在不违背本说明书内涵的情况下做类似推广，因此本说明书不受下面公开的具体实施的限制。
[0036]在本说明书一个或多个实施例中使用的术语是仅仅出于描述特定实施例的目的，而非旨在限制本说明书一个或多个实施例。在本说明书一个或多个实施例和所附权利要求书中所使用的单数形式的“一种”、“所述”和“该”也旨在包括多数形式，除非上下文清楚地表示其他含义。还应当理解，本说明书一个或多个实施例中使用的术语“和/或”是指并包含一个或多个相关联的列出项目的任何或所有可能组合。
[0037]应当理解，尽管在本说明书一个或多个实施例中可能采用术语第一、第二等来描述各种信息，但这些信息不应限于这些术语。这些术语仅用来将同一类型的信息彼此区分开。例如，在不脱离本说明书一个或多个实施例范围的情况下，第一也可以被称为第二，类似地，第二也可以被称为第一。取决于语境，如在此所使用的词语“如果”可以被解释成为“在
……
时”或“当
……
时”或“响应于确定”。
[0038]首先，对本说明书一个或多个实施例涉及的名词术语进行解释。
[0039]Wasserstein Distance：一种常见的度量分布之间距离的方法
[0040]旋转目标检测是计算机视觉中的一项基本任务，相比于通用的目标检测而言，其检测的精度更高，在遥感、文字检测等场景中有着广泛的用途。旋转目标检测通常有两种做法，一种是用一个旋转矩形框去表示目标的位置，另一种是用一个多边形框去表示目标的位置。不管用哪种方法本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种检测模型训练方法，包括：获取标注有目标框的样本图像；将所述样本图像输入预设检测模型中进行目标检测，获得预测框；根据所述目标框的边信息和点信息，以及所述预测框的边信息和点信息，计算损失值；基于所述损失值调整所述预设检测模型的模型参数，返回执行所述获取标注有目标框的样本图像的步骤，在达到预设训练停止条件的情况下，确定训练后的所述预设检测模型为目标检测模型。2.根据权利要求1所述的方法，所述边信息包括边的方向向量，所述点信息包括指定点的坐标；所述根据所述目标框的边信息和点信息，以及所述预测框的边信息和点信息，计算损失值，包括：根据所述目标框中各边的方向向量和所述预测框中各边的方向向量，确定方向差异；根据所述目标框中指定点的坐标和所述预测框中所述指定点的坐标，确定坐标差异；根据所述方向差异和所述坐标差异，得到损失值。3.根据权利要求2所述的方法，所述根据所述目标框中各边的方向向量和所述预测框中各边的方向向量，确定方向差异，包括：根据所述目标框中指定边的方向向量与所述预测框中所述指定边的方向向量，确定所述指定边对应的子方向差异，其中，所述指定边为任意一条边；根据各所述指定边对应的子方向差异，计算得到方向差异。4.根据权利要求2或3所述的方法，所述目标框和所述预测框中包括多个指定点；所述根据所述目标框中指定点的坐标和所述预测框中所述指定点的坐标，确定坐标差异，包括：根据所述目标框中第一指定点的坐标与所述预测框中所述第一指定点的坐标，确定所述第一指定点对应的子坐标差异，其中，所述第一指定点为任意一个指定点；根据各所述第一指定点对应的子坐标差异，计算得到坐标差异。5.根据权利要求2或3所述的方法，所述指定点为所述边的中点。6.根据权利要求2或3所述的方法，所述根据所述目标框中各边的方向向量和所述预测框中各边的方向向量，确定方向差异之前，还包括：获取所述目标框的点序列和所述预测框的点序列；根据所述目标框的点序列，确定所述目标框中各边的方向向量和指定点的坐标，并根据所述预测框的点序列，确定所述预测框中各边的方向向量和指定点的坐标。7.根据权利要求6所述的方法，所述根据所述目标框的点序列，确定所述目标框中各边的方向向量和指定点的坐标，并根据所述预测框的点序列，确定所述预测框中各边的方向向量和指定点的坐标，包括：根据所述目标框的点序列，确定所述目标框中各边的方向角、长度和中点，并根据所述预测框的点序列，确定所述预测框中各边的方向角、长度和中点；根据所述目标框中各边的方向角和长度，确定所述目标框中各边的方向向量，并根据所述预测框中各边的方向角和长度，确定所述预测框中各边的方向向量；将所述目标框中各边中点的坐标确定为目标框中指定点的坐标，将所述预测框中各边
中点的坐标确定为预测框中指定点的坐标。8.根据权利要求1所述的方法，所述目标框为目标矩形框，所述预测框为预测矩形框，所...

【专利技术属性】
技术研发人员：朱禹轲，阮宇艨，张悦，郭胜，韩冰，
申请(专利权)人：浙江网商银行股份有限公司，
类型：发明
国别省市：