[发明专利]选择器模型的训练方法、图像检测方法及相关设备

说明书

本申请适用于人工智能领域，提供了选择器模型的训练方法、图像检测方法及相关设备。选择器模型的训练方法包括：获取奖励函数和图像检测条件；执行训练步骤，训练步骤包括：根据奖励函数和图像检测条件对选择器模型进行训练，得到待优化的选择器模型；根据待优化的选择器模型和图像检测条件对奖励函数进行训练，得到待优化的奖励函数。将待优化的奖励函数作为新的奖励函数，将待优化的选择器模型作为新的选择器模型，循环执行训练步骤，直到得到目标奖励函数以及目标选择器模型。其中，目标奖励函数和目标选择器模型满足预设训练停止条件，从而提高选择器模型的训练精度，进而根据训练好的目标选择器模型选择出更准确的图像检测器。

技术领域

本申请属于人工智能技术领域，尤其涉及选择器模型的训练方法、图像检测方法及相关设备。

背景技术

现有的图像检测系统，在模型训练过程中，一般是固定奖励函数对模型进行训练，得到最优的模型。但是不同的图像检测条件，例如，不同的硬件运行环境、不同的图像检测精度等对模型训练过程中的奖励函数的要求不同，在多变的图像检测条件下，固定的奖励函数无法保证模型训练的精度。

发明内容

有鉴于此，本申请实施例提供了选择器模型的训练方法、图像检测方法及相关设备，以在选择器模型的训练过程中提高模型训练的精度，进而选择出准确的图像检测器进行图像检测。

本申请实施例的第一方面提供了一种选择器模型的训练方法，所述选择器模型是用于根据图像检测条件选择对应的图像检测器的标识的模型，所述图像检测器用于检测图像，所述选择器模型的训练方法包括：

获取奖励函数和图像检测条件；

执行训练步骤，所述训练步骤包括：

根据所述奖励函数和所述图像检测条件对所述选择器模型进行训练，得到待优化的选择器模型；

根据所述待优化的选择器模型和所述图像检测条件对所述奖励函数进行训练，得到待优化的奖励函数；

将所述待优化的奖励函数作为新的奖励函数，将所述待优化的选择器模型作为新的选择器模型，循环执行所述训练步骤，直到满足预设训练停止条件，将满足预设训练停止条件的待优化的奖励函数作为目标奖励函数，将满足预设训练停止条件的待优化的选择器模型作为目标选择器模型。

在一种可能的实现方式中，所述根据所述奖励函数和所述图像检测条件对所述选择器模型进行训练，得到待优化的选择器模型，包括：

根据所述图像检测条件和所述选择器模型确定第一标识；

根据确定的第一标识对应的图像检测器和预设的测试图像计算所述奖励函数的取值；

根据所述奖励函数的取值调整所述选择器模型的参数，得到奖励函数的取值大于预设值时对应的选择器模型，根据奖励函数的取值大于预设值时对应的选择器模型的参数生成所述待优化的选择器模型。

在一种可能的实现方式中，所述根据确定的第一标识对应的图像检测器和预设的测试图像计算所述奖励函数的取值，包括：

根据所述确定的第一标识对应的图像检测器检测所述测试图像；

根据所述测试图像的检测结果计算所述奖励函数的取值。

在一种可能的实现方式中，所述根据所述待优化的选择器模型和所述图像检测条件对所述奖励函数进行训练，得到待优化的奖励函数，具体包括：

根据所述待优化的选择器模型和所述图像检测条件确定第二标识；

根据确定的第二标识对应的图像检测器和预设的测试图像调整奖励函数的参数，得到取值大于预设值的奖励函数，根据取值大于预设值的奖励函数的参数生成所述待优化的奖励函数。