[发明专利]一种图像检索模型、训练方法

权利要求书

1.一种图像检索模型训练方法，其特征在于，该训练方法将该模型的损失函数L_RMS设为，

其中，符号参数说明如下：

i、k分别代表第i个、第k个样本；

m是全部样本的数量；

P为正样本对集合；

P_i为第i个样本作查询样本时，正样本对的集合；

N为负样本对集合；

N_i为第i个样本作查询样本时，负样本对的集合；

n为选择的负样本对数量；

δ为负样本对相似度从高到底排序的顺序(1-n)；

r为阈值参数，代表对负样本对优化的权重；

λ、d为阈值参数，共同表示将正负样本拉开的距离(相似度差距)；

S表示相似度；

S_ik表示第i个样本(查询样本)和第k的样本的相似度；

α、β为阈值参数，分别代表对正样本对损失的权重和对负样本对损失的权重。

2.根据权利要求1所述的图像检索模型训练方法，其特征在于，对于所述式(8)，令

其中的S_ik为

其中，r为比例系数，n为所有图像检索模型训练中负样本的数量。

3.根据权利要求1所述的图像检索模型训练方法，其特征在于，对于所述式(8)，令

正样本对的相似度为

负样本对的相似度为

4.一种图像检索模型，其特征在于，所述模型的训练方法如权利要求1所述。

5.一种图像检索装置，其特征在于，所述装置包括存储器；以及

耦合到所述存储器的处理器，该处理器被配置为执行存储在所述存储器中的指令，所述处理器通过图像检索模型对图像进行检索，该图像检索模型的训练方法如权利要求1所述。

6.一种存储介质，其上存储有计算机程序，其特征在于，该程序被处理器执行时，实现如权利要求1至3中任一所述的方法。