[發明專利]加快圖像檢索的方法、裝置、計算機設備和存儲介質

在審

• 申請號: 202010064425.4
• 申請日: 2020-01-20
• 公開（公告）號: CN111382299A
• 公開（公告）日: 2020-07-07
• 發明（設計）人: 陳旋;王沖;崇傳兵
• 申請（專利權）人: 江蘇艾佳家居用品有限公司
• 主分類號: G06F16/583
• 分類號: G06F16/583;G06F16/55;G06K9/62
• 代理公司: 南京正聯知識產權代理有限公司 32243
• 代理人: 鄧唯

1.一種加快圖像檢索的方法，其特征在于，包括如下步驟：

S10，獲取多個參考圖像的特征向量，將各個特征向量分別作為一個初始類；所述初始類為只有一個節點的樹；

S20，將特征向量的余弦相似度最大的兩個初始類確定為一個中間類，將中間類對應的兩個特征向量的平均向量確定為中間類根節點的特征向量，將中間類對應的兩個初始類作為中間類根節點的兩個子樹；

S30，重復執行步驟S20，直至只剩下一個初始類為止，根據各個中間類以及剩下一個初始類確定包括所有參考圖像的特征向量的聚類二叉樹；

S40，獲取目標圖像的特征向量，得到目標特征向量，在聚類二叉樹查找與目標特征向量之間的余弦相似度差距最小的中間類根節點，得到目標根節點，在目標根節點的子樹中查找與目標特征向量之間的余弦相似度差距最小的子節點，將查找得到的子節點的圖像確定為與目標圖像最相似的圖像。

2.根據權利要求1所述的加快圖像檢索的方法，其特征在于，在一個實施方式中，獲取多個參考圖像的特征向量包括：

采用VGG16提取各個參考圖像對應的最后一層卷積特征作為相應參考圖像的特征向量。

3.根據權利要求1所述的加快圖像檢索的方法，其特征在于，在一個實施方式中，兩個參考圖像之間的余弦相似度的確定公式包括：

Q₁＝[x₁，x₂，x₃，...，x₅₁₂]，

Q₂＝[y₁，y₂，y₃，...，y₅₁₂]，

其中，cos(θ)表示余弦相似度，Q₁表示一個參考圖像的特征向量，Q₂表示另一個參考圖像的特征向量，n的取值為512。

4.一種加快圖像檢索的裝置，其特征在于，包括：

第一獲取模塊，用于獲取多個參考圖像的特征向量，將各個特征向量分別作為一個初始類；所述初始類為只有一個節點的樹；

確定模塊，用于將特征向量的余弦相似度最大的兩個初始類確定為一個中間類，將中間類對應的兩個特征向量的平均向量確定為中間類根節點的特征向量，將中間類對應的兩個初始類作為中間類根節點的兩個子樹；

執行模塊，用于進入確定模塊重復執行相應的過程，直至只剩下一個初始類為止，根據各個中間類以及剩下一個初始類確定包括所有參考圖像的特征向量的聚類二叉樹；

第二獲取模塊，用于獲取目標圖像的特征向量，得到目標特征向量，在聚類二叉樹查找與目標特征向量之間的余弦相似度差距最小的中間類根節點，得到目標根節點，在目標根節點的子樹中查找與目標特征向量之間的余弦相似度差距最小的子節點，將查找得到的子節點的圖像確定為與目標圖像最相似的圖像。

5.根據權利要求4所述的加快圖像檢索的裝置，其特征在于，在一個實施方式中，所述第一獲取模塊進一步用于：

采用VGG16提取各個參考圖像對應的最后一層卷積特征作為相應參考圖像的特征向量。

6.根據權利要求4所述的加快圖像檢索的裝置，其特征在于，在一個實施方式中，兩個參考圖像之間的余弦相似度的確定公式包括：

Q₁＝[x₁，x₂，x₃，...，x₅₁₂]，

Q₂＝[y₁，y₂，y₃，...，y₅₁₂]，

其中，cos(θ)表示余弦相似度，Q₁表示一個參考圖像的特征向量，Q₂表示另一個參考圖像的特征向量，n的取值為512。

7.一種計算機設備，包括存儲器、處理器及存儲在存儲器上并可在處理器上運行的計算機程序，所述處理器執行所述計算機程序時實現權利要求1至6任一項所述的加快圖像檢索的方法的步驟。

8.一種計算機可讀存儲介質，其上存儲有計算機程序，所述計算機程序被處理器執行時實現權利要求1至6任一項所述的增強現實交互方法的步驟。