[发明专利]图像处理模型的确定方法、装置、设备及存储介质

说明书

本公开关于一种图像处理模型的确定方法、装置、设备及存储介质，涉及图像处理领域。本公开实施例中，对于每个样本图像，通过增加旋转图像，考虑到旋转图像的哈希码与样本图像的哈希码一致的特性，引入了另一种损失值，结合这种损失值与训练样本图像之间相似度的损失值，对图像处理模型的模型参数进行更新，能够使得图像处理模型在获取到更准确的哈希码，基于哈希码进行后续处理，也能够得到更准确的处理结果，因而能够提高该图像处理模型的准确性，且旋转图像为对样本图像进行几何变换得到的，这样该图像处理模型对几何变换前后的图像均能够进行准确处理，具有更好的鲁棒性，且无需标注数据，减少了人工成本，提高数据处理效率。

技术领域

本公开涉及图像处理领域，尤其涉及图像处理模型的确定方法、装置、设备及存储介质。

背景技术

在图像处理领域，能够通过图像处理模型对两个图像进行处理，以确定出两个图像的哈希码，进而根据哈希码确定出两个图像之间的相似度，来确定两个图像是否相关，以此来作为图像检索的依据。

目前，图像处理模型的训练过程通常需要为图像添加标注数据，来指导图像处理模型的训练过程。或者通过对样本图像进行特征提取，直接用图像对中两个样本图像的图像特征之间的相似度来作为训练目标，以指导图像处理模型的训练过程。相似度值越大，则认为两个样本图像之间越相似，后续计算两个样本图像之间的哈希码之间的距离则就越小。

但上述标注数据一般依赖大量的人工标注，人工成本较高，且非常耗时，因而，图像处理效率很低。直接使用图像特征之间的相似度来作为训练目标，由于图像特征得到的相似度是嘈杂的，并不能代表真实的相似关系，因而训练得到的图像处理模型的准确性较差。

发明内容

本公开提供一种图像处理模型的确定方法、装置、设备及存储介质，提高了图像处理的准确性。本公开的技术方案如下：

根据本公开实施例的第一方面，提供一种图像处理模型的确定方法，包括：

获取多个样本图像和所述多个样本图像的旋转图像，每个样本图像的旋转图像为对所述样本图像旋转目标角度得到的图像；

基于图像处理模型，对所述多个样本图像和所述多个样本图像的旋转图像进行处理，得到所述多个样本图像的相似度信息、所述多个样本图像的哈希码和所述多个样本图像的旋转图像的哈希码，所述相似度信息包括所述多个样本图像中两个样本图像之间的相似度；

基于所述相似度信息中两个样本图像之间的相似度以及所述两个样本图像的哈希码之间的差距，获取第一损失值；

基于所述多个样本图像中每个样本图像的哈希码和所述样本图像的旋转图像的哈希码之间的差距，获取第二损失值；

基于所述第一损失值和所述第二损失值，对所述图像处理模型的模型参数进行更新，直至符合目标条件时停止，得到目标图像处理模型。

在一些实施例中，所述基于图像处理模型，对所述多个样本图像和所述多个样本图像的旋转图像进行处理，得到所述多个样本图像的相似度信息、所述多个样本图像的哈希码和所述多个样本图像的旋转图像的哈希码，包括：

基于图像处理模型，对所述多个样本图像和所述多个样本图像的旋转图像进行特征提取，得到所述多个样本图像的图像特征和所述多个样本图像的旋转图像的图像特征；

基于所述图像处理模型以及所述多个样本图像的图像特征，获取所述多个样本图像的相似度信息以及所述多个样本图像的哈希码；

基于所述图像图像处理模型和所述多个样本图像的旋转图像的图像特征，获取所述旋转图像的哈希码。

在一些实施例中，所述基于所述图像处理模型以及所述多个样本图像的图像特征，获取所述多个样本图像的相似度信息，包括：