[發明專利]基于域控制器的圖像檢索方法及計算機可讀存儲介質

在審

• 申請號: 202211098261.2
• 申請日: 2022-09-08
• 公開（公告）號: CN115375933A
• 公開（公告）日: 2022-11-22
• 發明（設計）人: 王達君;梁俊文
• 申請（專利權）人: 重慶中科云從科技有限公司
• 主分類號: G06V10/762
• 分類號: G06V10/762;G06V10/74;G06V10/82
• 代理公司: 北京瀚仁知識產權代理事務所(普通合伙) 11482
• 代理人: 陳敏

1.一種基于域控制器的圖像檢索方法，所述域控制器包括多個計算單元，其特征在于，所述方法包括：

根據目標檢索圖像集的聚類類別對應的聚類中心總數量、第一相似度單位計算時長、預設鄰近子類數量、每個所述聚類類別對應的標準特征向量數量以及第二相似度單位計算時長將所述多個計算單元劃分為第一計算單元和第二計算單元；

利用訓練好的特征提取網絡對待查詢圖像進行特征提取，得到所述待查詢圖像的特征向量；

利用所述第一計算單元計算所述待查詢圖像的特征向量和所述聚類類別對應的聚類中心的第一相似度，并根據所述第一相似度和所述預設鄰近子類數量從所述聚類類別中確定出所述待查詢圖像的鄰近子類別；

利用所述第二計算單元計算所述待查詢圖像的特征向量和所述鄰近子類別對應的標準特征向量的第二相似度，并根據所述第二相似度得到所述待查詢圖像的檢索結果。

2.根據權利要求1所述的方法，其特征在于，所述根據目標檢索圖像集的聚類類別對應的聚類中心總數量、第一相似度單位計算時長、預設鄰近子類數量、每個所述聚類類別對應的標準特征向量數量以及第二相似度單位計算時長將所述多個計算單元劃分為第一計算單元和第二計算單元，包括：

根據所述目標檢索圖像集的所述聚類類別對應的所述聚類中心總數量和所述第一相似度單位計算時長，確定第一計算資源量；

根據每個所述聚類類別對應的所述標準特征向量數量計算聚類類別平均標準特征向量數量；根據所述預設鄰近子類數量、所述聚類類別平均標準特征向量數量和所述第二相似度單位計算時長，確定第二計算資源量；

根據所述第一計算資源量和所述第二計算資源量將所述多個計算單元劃分為所述第一計算單元和所述第二計算單元。

3.根據權利要求2所述的方法，其特征在于，所述根據所述第一計算資源量和所述第二計算資源量將所述多個計算單元劃分為所述第一計算單元和所述第二計算單元，包括：

根據所述第一計算資源量和所述第二計算資源量的比值，確定所述第一計算單元和/或所述第二計算單元在所述多個計算單元中的占比；

根據所述占比將所述多個計算單元劃分為所述第一計算單元和所述第二計算單元。

4.根據權利要求1所述的方法，其特征在于，所述域控制器還包括與每個計算單元對應配置的局部存儲器，所述方法還包括預先將所述目標檢索圖像集的聚類信息存儲至所述局部存儲器，其中，所述聚類信息包括所述聚類類別以及與所述聚類類別對應的所述聚類中心和所述標準特征向量；

利用所述第一計算單元計算所述待查詢圖像的特征向量和所述聚類類別對應的聚類中心的第一相似度包括：

從與所述第一計算單元對應的所述局部存儲器中讀取所述聚類中心，并利用所述第一計算單元計算所述待查詢圖像的特征向量和所述聚類中心的第一相似度。

5.根據權利要求1至4中任意一項所述的方法，其特征在于，所述根據所述第一相似度和所述預設鄰近子類數量從所述聚類類別中確定出所述待查詢圖像的鄰近子類別，包括：

從所述第一計算單元中讀取所述第一相似度，將所述第一相似度由小到大進行排序；根據與排序靠前的所述預設鄰近子類數量的所述第一相似度對應的所述聚類中心確定所述待查詢圖像的所述鄰近子類別；

和/或，

所述根據所述第二相似度得到所述待查詢圖像的檢索結果，包括：

從所述第二計算單元中讀取所述第二相似度，將所述第二相似度由小到大進行排序；

根據與排序靠前的預設鄰近特征數量的所述第二相似度對應的所述標準特征向量得到所述待查詢圖像的檢索結果。