[发明专利]模型训练及目标检测方法、装置、存储介质及电子设备

说明书

本说明书公开了模型训练及目标检测方法、装置、存储介质及电子设备。在建立标注框和参考框之间的对应关系时，可以先根据标注框和参考框的位置为标注框筛选出一部分与该标注框距离较近的参考框，并仅为标注框建立与所筛选出的参考框之间的对应关系。由于确定标注框和参考框的位置以及确定标注框和参考框之间的距离所需要消耗的计算资源相较于计算标注框和参考框之间的重合度所消耗的计算资源较小，因此，采用本说明书提供的模型训练方法能够在占用较少的计算资源的条件下更快地建立标注框和参考框之间的对应关系。

技术领域

本说明书涉及计算机技术领域，尤其涉及模型训练及目标检测方法、装置、存储介质及电子设备。

背景技术

目前，通过一些目标检测模型，可以采用基于参考框的算法实现目标检测。

在基于参考框的目标检测算法的思想中，可以基于预先锚定出的参考框来预测检测框，例如，可以通过确定基于参考框的位置偏移，来定位出检测框。

在对目标检测模型的训练过程中，通常会在所要输入模型的样本图像中标注出标注框，并生成若干个参考框。根据目标检测任务的需要，标注框的标签可以包括标注框的位置、尺寸，以及标注框所框定的子图像的语义等等。然后，为每个标注框筛选出与该标注框之间的重合度大于指定的重合度的参考框，并建立该标注框与所筛选出的各参考框之间的对应关系。

当所输出的检测框是基于某一参考框得到的，则根据定位该检测框所基于的参考框对应的标注框与该检测框之间的差异，构建用于训练所述目标检测模型的差异项。

但在确定参考框与标注框之间的对应关系时，针对每个标注框，都需要计算该标注框与所有参考框之间的重合度，从而为与标注框之间重合度较大的参考框建立与标注框之间的对应关系，而重合度的计算会消耗大量的计算资源。

发明内容

本说明书提供模型训练及目标检测方法、装置、存储介质及电子设备，以部分的解决现有技术存在的上述问题。

本说明书采用下述技术方案：

本说明书提供了一种模型训练方法，包括：

确定样本图像，并将样本图像输入至待训练的目标检测模型，得到所述样本图像对应的特征图；

在样本图像上生成若干个参考框，针对每个参考框，根据该参考框所框定出的子图像在特征图上对应的特征区域，通过所述目标检测模型，输出基于该参考框的识别结果，作为该参考框对应的识别结果；

基于所建立的参考框与样本图像上的标注框之间的对应关系，针对每个参考框，以该参考框对应的识别结果和该参考框对应的标注框的标注结果之间的差异最小为目标，调整所述目标检测模型中的参数；

其中，建立参考框与样本图像上的标注框之间的对应关系，具体包括：

确定各参考框和各标注框在样本图像中的位置；

针对每个标注框，从各参考框中筛选出至少部分与该标注框之间的距离小于指定距离的参考框，并建立该标注框与所筛选出的参考框之间的对应关系。

可选地，输出基于该参考框的识别结果，具体包括：

输出预测框相对于该参考框的位置偏移；或，

输出基于该参考框所确定出的预测框所框定出的子图像中所包含的对象的类别。

可选地，从各参考框中筛选出至少部分与该标注框之间的距离小于指定距离的参考框，具体包括：

确定该标注框的中心点以及所述筛选出的各参考框的中心点；