[发明专利]用于训练模型的方法、装置、设备、介质和程序产品

权利要求书

1.一种图像处理方法，包括：

获取输入的图像数据；

利用经训练的第一模型确定针对所述输入的图像数据的预测结果，所述经训练的第一模型是图像分类模型、语义分割模型以及目标识别模型中的一项，并且所述预测结果是所述输入的图像数据的分类结果、语义分割结果、目标识别结果中的对应的一项，

其中，经训练的所述第一模型通过如下过程而得到：

获取第一模型和第二模型，其中所述第一模型至少包括一个或多个第一层级组，所述第一层级组包括第一特征提取层和第一降采样层，所述第二模型包括一个或多个第二层级组，所述第二层级组包括第二特征提取层、第三特征提取层和第二降采样层，所述第一模型和所述第二模型被初始化以具有不同的模型参数；

基于图像训练样本，根据所述第一模型和第二模型来确定第一约束，所述图像训练样本具有指示所述图像训练样本的类别的标签，所述第一约束与所述第一模型和第二模型中的对应层级组中的所述第一降采样层和所述第二降采样层针对所述图像训练样本的输出之间的差异有关，所述第一降采样层的输出和所述第二降采样层的输出具有相同的维度；

基于所述图像训练样本，根据第一模型来确定第二约束，所述第二约束与所述第一模型针对所述图像训练样本的输出和所述标签的差异有关；

基于所述图像训练样本，根据第二模型来确定第三约束，所述第三约束与所述第二模型针对所述图像训练样本的输出和所述标签的差异有关；以及

至少基于所述第一约束、所述第二约束和所述第三约束，通过调整所述第一模型和所述第二模型的模型参数来训练所述第一模型和所述第二模型，以获取经训练的所述第一模型，

其中，所述图像训练样本包括原始样本和增广样本，所述增广样本通过对所述原始样本中的图像进行裁剪、旋转或翻转而获得。

2.根据权利要求1所述的方法，其中基于所述图像训练样本，根据所述第一模型和第二模型来确定所述第一约束包括：

确定所述对应层级组中的所述第一降采样层针对所述图像训练样本输出的第一特征图；

确定所述对应层级组中的所述第二降采样层针对所述图像训练样本输出的第二特征图；以及

基于所述第一特征图和所述第二特征图之间的差异，来确定所述第一约束。

3.根据权利要求1所述的方法，其中至少基于所述第一约束，训练所述第一模型和所述第二模型，以获取经训练的所述第一模型包括：

基于所述第一约束、所述第二约束和所述第三约束以及与其相关联的权重，来训练所述第一模型和所述第二模型，以获取经训练的所述第一模型。