[發明專利]一種圖像檢索模型、訓練方法

在審

• 申請號: 202110365966.5
• 申請日: 2021-04-06
• 公開（公告）號: CN113326864A
• 公開（公告）日: 2021-08-31
• 發明（設計）人: 王建;張智超;宋巍;黃冬梅;趙丹楓;鄭小羅
• 申請（專利權）人: 上海海洋大學
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06N20/00
• 代理公司: 上海伯瑞杰知識產權代理有限公司 31227
• 代理人: 孟旭彤

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中任一所述的方法。