[发明专利]目标检测网络训练中候选框匹配方法、装置、设备及介质

权利要求书

1.一种目标检测网络训练中候选框匹配方法，其特征在于，包括：

获取包含原始图像的目标检测训练数据集，在所述原始图像中设置初始候选框；

对预定的训练度字典执行初始化操作，以便将所述训练度字典中每个初始候选框对应的训练度设置为初始值；

将包含目标框的图像输入到目标检测网络中进行训练，并将所述初始候选框与所述目标框进行匹配，根据匹配结果对所述训练度字典进行更新；

在所述目标检测网络的每一轮训练中，从更新后的所述训练度字典中选择预设数量的低训练度候选框，将所述低训练度候选框与待匹配的目标框进行裁剪匹配，根据裁剪匹配结果对所述训练度字典进行更新；

将裁剪后的待匹配的目标框对应的图像输入到所述目标检测网络中进行训练，直至所述目标检测网络的训练轮数达到预设要求。

2.根据权利要求1所述的方法，其特征在于，所述对预定的训练度字典执行初始化操作，包括：

根据所述初始候选框的数量对所述训练度字典中训练度对应的字符长度进行设置，并将所述训练度字典中每个所述初始候选框对应的训练度的初始值设置为0。

3.根据权利要求1所述的方法，其特征在于，所述将包含目标框的图像输入到目标检测网络中进行训练，并将所述初始候选框与所述目标框进行匹配，包括：

获取批量图像，所述批量图像中包含多个目标框，将所述批量图像输入到预先搭建的目标检测网络中进行训练；

在每一轮训练中，计算所述初始候选框与所述目标框之间的交并比，当所述交并比大于预设阈值时，将所述交并比大于预设阈值时的所述初始候选框和所述目标框匹配，并将所述交并比大于预设阈值的初始候选框对应的训练度的值增加第一分值；

其中，所述初始候选框可以匹配一个或多个所述目标框，每当所述初始候选框匹配上一个所述目标框时，均在所述训练度字典中将所述初始候选框对应的训练度的值增加第一分值。

4.根据权利要求1所述的方法，其特征在于，所述从更新后的所述训练度字典中选择预设数量的低训练度候选框，将所述低训练度候选框与待匹配的目标框进行裁剪匹配，包括：

在更新后的所述训练度字典中按照所述初始候选框对应的训练度的值进行排序，并根据排序结果选出预设数量的低训练度候选框；

对所述待匹配的目标框对应的图像进行缩放和裁剪，将裁剪出来的所述待匹配的目标框的图像粘贴到所述低训练度候选框上，并使所述待匹配的目标框与所述低训练度候选框之间的交并比为50％-100％；

利用粘贴后的图像对所述目标检测网络进行训练，在每一轮训练中，将匹配到所述待匹配的目标框的所述低训练度候选框对应的训练度的值增加第二分值。

5.根据权利要求1所述的方法，其特征在于，在所述根据裁剪匹配结果对所述训练度字典进行更新之后，所述方法还包括：

从更新后的所述训练度字典中选择预设数量的校准候选框，并对所述原始图像进行遍历，选取混融基图和混融附图；

对所述混融附图进行随机缩放，以便将所述混融附图中的目标框的大小缩放到所述校准候选框大小的80％-120％；

将随机缩放后的混融附图与所述混融基图相加得到混融图像，并使所述目标框与所述校准候选框之间的交并比为50％-100％；

将所述混融图像输入到所述目标检测网络中对所述校准候选框进行训练，并将训练中匹配到所述目标框的所述校准候选框对应的训练度的值增加第三分值。

6.根据权利要求5所述的方法，其特征在于，所述从更新后的所述训练度字典中选择预设数量的校准候选框，并对所述原始图像进行遍历，选取混融基图和混融附图，包括：

基于所述更新后的所述训练度字典中训练度的排序结果，由低到高选取预设数量的低训练度候选框作为所述校准候选框；

遍历所述目标检测训练数据集中的原始图像，每一次遍历选取一张原始图像作为混融基图，并随机选取其他两张原始图像作为混融附图。