[發明專利]一種圖片搜索模型的選擇方法和裝置

在審

• 申請號: 202210001497.3
• 申請日: 2022-01-04
• 公開（公告）號: CN114332508A
• 公開（公告）日: 2022-04-12
• 發明（設計）人: 王事成
• 申請（專利權）人: 北京易華錄信息技術股份有限公司
• 主分類號: G06V10/74
• 分類號: G06V10/74;G06V10/774;G06K9/62;G06N20/00
• 代理公司: 北京智沃律師事務所 11620
• 代理人: 吳志宏

1.一種圖片搜索模型的選擇方法，其特征在于，包括以下步驟：

設置參考圖片庫和待檢索圖片集，所述參考圖片庫包括多張參考圖片，所述待檢索圖片集包括多張待檢索圖片，且所述參考圖片庫中存在與每張待檢索圖片匹配的目標圖片；

使用手工特征模型從所述參考圖片庫中檢索與每張待檢索圖片相似度最高的預設數量的參考圖片；

根據所述預設數量的參考圖片，獲取所述手工特征模型的評測指標；

在所述手工特征模型的評測指標滿足預設條件的情況下，選擇所述手工特征模型作為圖片搜索模型；

在所述手工特征模型的評測指標不滿足預設條件的情況下，使用深度學習模型從所述參考圖片庫中檢索與每張待檢索圖片相似度最高的預設數量的參考圖片；

根據所述預設數量的參考圖片，獲取所述深度學習模型的評測指標；

在所述深度學習模型的評測指標滿足預設條件的情況下，選擇所述深度學習模型作為圖片搜索模型。

2.根據權利要求1所述的方法，其特征在于，所述使用手工特征模型從所述參考圖片庫中檢索與每張待檢索圖片相似度最高的預設數量的參考圖片，具體包括：

使用手工特征模型生成所述參考圖片庫中的每張參考圖片和所述待檢索圖片集中的每張待檢索圖片的特征編碼；

計算每張待檢索圖片的特征編碼與所述參考圖片庫中的每張參考圖片的特征編碼之間的歐式距離；

根據所述歐式距離，確定與每張待檢索圖片相似度最高的預設數量的參考圖片。

3.根據權利要求1所述的方法，其特征在于，所述評測指標包括命中率和得分；

所述根據所述預設數量的參考圖片，獲取所述手工特征模型的評測指標，具體包括：

根據與每張待檢索圖片相似度最高的預設數量的參考圖片中是否包括與該待檢索圖片對應的目標圖片，確定所述手工特征模型的命中率；

在與每張待檢索圖片相似度最高的預設數量的參考圖片中包括與該待檢索圖片對應的目標圖片的情況下，根據所述目標圖片在所述預設數量的參考圖片中的相似度排序，確定所述手工特征模型的得分。

4.根據權利要求1所述的方法，其特征在于，所述根據所述預設數量的參考圖片，獲取所述深度學習模型的評測指標之后，還包括：

在所述深度學習模型的評測指標不滿足預設條件的情況下，根據用戶在參數面板選擇的圖片變換類型，更新訓練圖片集；其中，所述圖片變換類型包括圖層覆蓋、空間變換、像素抖動和顏色變換；

基于更新后的訓練圖片集，對所述手工特征模型和所述深度學習模型進行在線訓練。

5.根據權利要求4所述的方法，其特征在于，所述圖層覆蓋包括文本覆蓋和圖片覆蓋；所述空間變換包括裁剪、旋轉、水平翻轉、豎向翻轉、填充、長寬比和透視變化；所述像素抖動包括模糊、編碼壓縮、銳化、馬賽克化和像素洗刷；所述顏色變換包括亮度、飽和度、對比度和灰度。

6.一種圖片搜索模型的選擇裝置，其特征在于，包括：

設置模塊，用于設置參考圖片庫和待檢索圖片集，所述參考圖片庫包括多張參考圖片，所述待檢索圖片集包括多張待檢索圖片，且所述參考圖片庫中存在與每張待檢索圖片匹配的目標圖片；

第一檢索模塊，用于使用手工特征模型從所述參考圖片庫中檢索與每張待檢索圖片相似度最高的預設數量的參考圖片；

第一獲取模塊，用于根據所述預設數量的參考圖片，獲取所述手工特征模型的評測指標；

第一選擇模塊，用于在所述手工特征模型的評測指標滿足預設條件的情況下，選擇所述手工特征模型作為圖片搜索模型；

第二檢索模塊，用于在所述手工特征模型的評測指標不滿足預設條件的情況下，使用深度學習模型從所述參考圖片庫中檢索與每張待檢索圖片相似度最高的預設數量的參考圖片；

第二獲取模塊，用于根據所述預設數量的參考圖片，獲取所述深度學習模型的評測指標；

第二選擇模塊，用于在所述深度學習模型的評測指標滿足預設條件的情況下，選擇所述深度學習模型作為圖片搜索模型。