[发明专利]图像处理模型的训练方法、图像相似度确定方法和装置

权利要求书

1.一种图像处理模型的训练方法，其中，所述图像处理模型包括特征提取网络和相似度确定网络；所述方法包括：

将样本图像中的第一图像输入所述特征提取网络，得到第一图像特征；

将所述样本图像中的第二图像输入所述特征提取网络，得到第二图像特征；其中，所述第二图像包括指示所述第一图像和所述第二图像之间的实际相似度的标签；

基于所述第一图像特征和所述第二图像特征，采用所述相似度确定网络，确定所述第一图像和所述第二图像之间的预测相似度；以及

根据所述预测相似度和所述实际相似度，对所述图像处理模型进行训练。

2.根据权利要求1所述的方法，所述基于所述第一图像特征和所述第二图像特征，采用所述相似度确定网络，确定所述第一图像和所述第二图像之间的预测相似度，包括：

根据所述第一图像特征与所述第二图像特征的点乘结果，确定所述预测相似度。

3.根据权利要求2所述的方法，其中，所述第二图像包括多个子图像；所述第二图像特征包括所述多个子图像中每个子图像的子图像特征；所述根据所述第一图像特征与所述第二图像特征的点乘结果，确定所述预测相似度包括：

根据所述第一图像特征与所述每个子图像的子图像特征的点乘结果，确定所述第一图像与所述每个子图像之间的预测相似度。

4.根据权利要求2所述的方法，其中，所述根据所述第一图像特征与所述第二图像特征的点乘结果，确定所述预测相似度，包括：

采用点乘运算确定所述第一图像特征与所述第二图像特征之间的距离特征；以及

对所述距离特征进行扁平化处理，得到所述预测相似度。

5.根据权利要求1所述的方法，其中：

所述第一图像包括目标字符；

所述第二图像包括多个子图像，所述多个子图像中的一部分图像包括与所述目标字符相似的第一字符，所述多个子图像中的另一部分图像包括与所述目标字符不相似的第二字符。

6.根据权利要求5所述的方法，还包括：

在模板图像上添加所述目标字符，得到所述第一图像；

获取所述目标字符的至少一个相似字符，并在所述模板图像上分别添加所述至少一个相似字符，得到所述一部分图像；以及

从预定字库中获取与所述目标字符不相似的至少一个字符，并在所述模板图像上分别添加所述至少一个字符，得到所述另一部分图像。

7.一种图像相似度确定方法，包括：

将多个待识别图像输入图像处理模型中的特征提取网络，得到第一特征矩阵；

将所述第一特征矩阵与所述第一特征矩阵的转置矩阵进行点乘运算，得到距离矩阵；以及

基于所述距离矩阵，确定所述多个待识别图像彼此之间的相似度；

其中，所述图像处理模型是采用权利要求1至6中任意一项所述的方法训练得到的。

8.一种图像处理模型的训练装置，其中，所述图像处理模型包括特征提取网络和相似度确定网络；所述装置包括：

第一特征确定模块，用于将样本图像中的第一图像输入所述特征提取网络，得到第一图像特征；

第二特征确定模块，用于将所述样本图像中的第二图像输入所述特征提取网络，得到第二图像特征；其中，所述第二图像包括指示所述第一图像和所述第二图像之间的实际相似度的标签；

预测模块，用于基于所述第一图像特征和所述第二图像特征，采用所述相似度确定网络，确定所述第一图像和所述第二图像之间的预测相似度；以及

模型训练模块，用于根据所述预测相似度和所述实际相似度，对所述图像处理模型进行训练。