[發(fā)明專利]一種索引樹建立方法以及圖像檢索方法

在審

• 申請?zhí)枺?/td>: 201810422259.3
• 申請日: 2018-05-04
• 公開（公告）號: CN108664583A
• 公開（公告）日: 2018-10-16
• 發(fā)明（設(shè)計(jì)）人: 謝明奇;顧嘉唯
• 申請（專利權(quán)）人: 北京物靈智能科技有限公司
• 主分類號: G06F17/30
• 分類號: G06F17/30
• 代理公司: 北京市天璽沐澤專利代理事務(wù)所(普通合伙) 11532
• 代理人: 謝鑫

1.一種索引樹建立方法，其特征在于，包括：

提取樣本圖像的樣本特征；

根據(jù)所述樣本特征，按提取順序建立初始索引；

根據(jù)所述初始索引、每個節(jié)點(diǎn)包含的最大子節(jié)點(diǎn)數(shù)、每個葉節(jié)點(diǎn)包含的最大索引數(shù)以及最大層數(shù)，建立索引樹。

2.如權(quán)利要求1所述的方法，其特征在于，所述索引樹的每個節(jié)點(diǎn)含有最大M個子節(jié)點(diǎn)或最大N個索引，N＝A/M^L，A為圖像特征總數(shù)，L為預(yù)先設(shè)定的最大層數(shù)。

3.如權(quán)利要求2所述的方法，其特征在于，根據(jù)所述初始索引、每個節(jié)點(diǎn)包含的最大子節(jié)點(diǎn)數(shù)、每個葉節(jié)點(diǎn)包含的最大索引數(shù)以及最大層數(shù)，建立索引樹，具體包括：

對所述初始索引按照偽隨機(jī)函數(shù)重新排列；

根據(jù)隨機(jī)排列后的初始索引、每個節(jié)點(diǎn)包含的最大子節(jié)點(diǎn)數(shù)、每個葉節(jié)點(diǎn)包含的最大索引數(shù)以及最大層數(shù)，建立索引樹。

4.如權(quán)利要求3所述的方法，其特征在于，根據(jù)隨機(jī)排列后的初始索引、每個節(jié)點(diǎn)包含的最大子節(jié)點(diǎn)數(shù)、每個葉節(jié)點(diǎn)包含的最大索引數(shù)以及最大層數(shù)，建立索引樹，具體包括：

從所述隨機(jī)排列后的初始索引中依次選取M個初始索引作為子節(jié)點(diǎn)，其中，M個子節(jié)點(diǎn)中任意兩個子節(jié)點(diǎn)的權(quán)重之間距離都不同，所述子節(jié)點(diǎn)的權(quán)重為所選取的初始索引所對應(yīng)的特征；

針對余下的每個初始索引，依次計(jì)算該初始索引對應(yīng)的特征與已遍歷到的M中的每個子節(jié)點(diǎn)的索引的權(quán)重之間的距離，確定與該初始索引對應(yīng)的特征距離最近的子節(jié)點(diǎn)，并將該初始索引劃分到所確定出的子節(jié)點(diǎn)對應(yīng)的初始索引集合中，其中，每個子節(jié)點(diǎn)均對應(yīng)一個初始索引集合；

對M個初始索引集合重復(fù)上述步驟直至所述索引樹達(dá)到L層或者所有圖像特征添加到索引樹。

5.如權(quán)利要求4所述的方法，其特征在于，該方法還包括：

若所述隨機(jī)排序后的初始索引中剩余的初始索引不夠M個，則認(rèn)為父節(jié)點(diǎn)為葉節(jié)點(diǎn)；

若子節(jié)點(diǎn)對應(yīng)的初始索引集合中的初始索引的個數(shù)小于或等于N，則此子節(jié)點(diǎn)為葉節(jié)點(diǎn)。

6.一種圖像檢索方法，其特征在于，所述圖像檢索方法基于上述索引樹建立方法，包括：

獲取待檢索的圖像；

提取待檢索的圖像的圖像特征；

根據(jù)所述圖像特征，在預(yù)先建立的索引樹中，匹配所述圖像特征對應(yīng)的索引，其中，所述索引樹具有預(yù)先設(shè)置的每個節(jié)點(diǎn)包含的最大子節(jié)點(diǎn)數(shù)、每個葉節(jié)點(diǎn)包含的最大索引數(shù)以及最大層數(shù)；

根據(jù)所述圖像特征對應(yīng)的索引，確定所述索引對應(yīng)的圖像。

7.如權(quán)利要求6所述的方法，其特征在于，根據(jù)所述圖像特征，在預(yù)先建立的索引樹中，匹配所述圖像特征對應(yīng)的索引，具體包括：

針對預(yù)先建立的索引樹的每層節(jié)點(diǎn)，根據(jù)所述圖像特征以及該層每個節(jié)點(diǎn)的權(quán)重，計(jì)算所述圖像特征與該層每個節(jié)點(diǎn)的權(quán)重之間距離，選取距離最小的節(jié)點(diǎn)，遍歷該節(jié)點(diǎn)對應(yīng)的下一層節(jié)點(diǎn)，其中，所述節(jié)點(diǎn)的權(quán)重為所述節(jié)點(diǎn)的索引所對應(yīng)的特征；

當(dāng)匹配到圖像特征在該層所對應(yīng)的節(jié)點(diǎn)為葉節(jié)點(diǎn)時，根據(jù)所述圖像特征，計(jì)算該葉節(jié)點(diǎn)的每個索引所對應(yīng)特征與所述圖像特征之間的距離，選取距離最小的索引作為所述圖像特征對應(yīng)的索引。

8.如權(quán)利要求7所述的方法，其特征在于，在選取距離最小的索引作為所述圖像特征對應(yīng)的索引之前，該方法還包括：在具有多個索引樹的情況下，比較所述圖像特征與每個索引樹的索引之間的距離，該距離中最小的距離對應(yīng)的索引作為所述圖像特征。

9.一種計(jì)算機(jī)可讀介質(zhì)，其特征在于，該計(jì)算機(jī)可讀介質(zhì)上存儲有程序，該程序被處理器執(zhí)行時實(shí)現(xiàn)根據(jù)權(quán)利要求1所述方法的步驟。

10.一種計(jì)算機(jī)系統(tǒng)，其特征在于，該計(jì)算機(jī)系統(tǒng)包括計(jì)算機(jī)可讀介質(zhì)和處理器；

計(jì)算機(jī)可讀介質(zhì)上存儲有程序；

當(dāng)所述程序被執(zhí)行時，所述處理器被配置成實(shí)現(xiàn)根據(jù)權(quán)利要求1所述方法的步驟。