[发明专利]图像识别模型的训练方法和装置

权利要求书

1.一种图像识别模型的训练方法，其特征在于，所述图像识别模型的训练方法包括：

获取第一样本图像和第二样本图像，所述第一样本图像包括第一样本对象，所述第一样本图像和所述第二样本图像属于不同图像域；

基于所述第一样本图像和所述第二样本图像，生成图像风格与所述第二样本图像相同的第一参考图像，所述第一参考图像包括所述第一样本对象；

将所述第一参考图像输入图像识别模型，通过所述图像识别模型对所述第一参考图像进行图像识别，输出所述第一样本对象在所述第一参考图像的第一目标位置；

若所述第一目标位置与所述第一样本对象在所述第一样本图像中位置之间的差异信息符合目标条件，将所述图像识别模型作为训练完成的图像识别模型。

2.根据权利要求1所述的方法，其特征在于，所述基于所述第一样本图像和所述第二样本图像，生成图像风格与所述第二样本图像相同的第一参考图像包括：

将所述第一样本图像输入所述图像识别模型，通过所述图像识别模型提取所述第一样本图像的第一样本内容特征；

将所述第二样本图像输入图像生成模型，通过所述图像生成模型提取所述第二样本图像的第二样本风格特征；根据所述第一样本内容特征和所述第二样本风格特征，生成所述第一参考图像。

3.根据权利要求1所述的方法，其特征在于，所述第二样本图像包括第二样本对象，所述方法还包括：

基于所述第一样本图像和所述第二样本图像，生成图像风格与所述第一样本图像相同的第二参考图像，所述第二参考图像包括所述第二样本对象；

对所述第二参考图像进行图像识别，输出所述第二样本对象在所述第二参考图像的第二目标位置；

根据所述第二目标位置与所述第二样本对象在所述第二样本图像中位置之间的差异信息，调整所述图像识别模型的模型参数。

4.根据权利要求1所述的方法，其特征在于，所述获取第一样本图像和第二样本图像之后，所述方法还包括：

将所述第一样本图像输入所述图像识别模型，通过所述图像识别模型提取所述第一样本图像的第一样本内容特征；

将所述第一样本图像输入图像生成模型，通过所述图像生成模型提取所述第一样本图像的第一样本风格特征；根据所述第一样本内容特征和所述第一样本风格特征，生成第三参考图像；

将所述第三参考图像输入所述图像识别模型，通过所述图像识别模型，对所述第三参考图像进行图像识别，输出所述第一样本对象在所述第三参考图像的第三目标位置；

根据所述第三目标位置与所述第一样本对象在所述第一样本图像中位置之间的差异信息，调整所述图像识别模型的模型参数。

5.根据权利要求1所述的方法，其特征在于，所述获取第一样本图像和第二样本图像之后，所述方法还包括：

将所述第一样本图像输入所述图像识别模型，通过所述图像识别模型提取所述第一样本图像的第一样本内容特征；

将所述第一样本图像输入图像生成模型，通过所述图像生成模型提取所述第一样本图像的第一样本风格特征；根据所述第一样本内容特征和所述第一样本风格特征，生成第三参考图像；

将所述第三参考图像输入所述图像识别模型，通过所述图像识别模型提取所述第三参考图像的第三参考内容特征；

根据所述第一样本内容特征和所述第三参考内容特征之间的差异信息，调整所述图像识别模型的模型参数。

6.根据权利要求5所述的方法，其特征在于，所述生成第三参考图像之后，所述方法还包括：

将所述第三参考图像输入所述图像生成模型，通过所述图像生成模型提取所述第三参考图像的第三参考风格特征；

根据所述第一样本风格特征和所述第三参考风格特征之间的差异信息，调整所述图像生成模型的模型参数。