[發明專利]圖像聚類方法、裝置電子設備、計算機可讀存儲介質

在審

• 申請號: 202011200839.1
• 申請日: 2020-11-02
• 公開（公告）號: CN112101483A
• 公開（公告）日: 2020-12-18
• 發明（設計）人: 張宏
• 申請（專利權）人: 浙江大華技術股份有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62
• 代理公司: 深圳市威世博知識產權代理事務所(普通合伙) 44280
• 代理人: 黎堅怡

本發明提供一種圖像聚類方法、裝置、電子設備及存儲介質，方法包括：獲取第一類簇的多個質心和第二類簇的多個質心；對比所述第一類簇的多個質心與第二類簇的多個質心的相似度，根據所述相似度的比較結果，確定所述第二類簇與所述第一類簇能否合并；若所述第二類簇與所述第一類簇能合并，則將所述第二類簇與所述第一類簇進行合并，得到合并后的類簇的質心；利用合并后的類簇的質心進行圖像聚類。以此解決聚類過程中數據漂移問題。

技術領域

本發明涉及圖像處理技術領域，尤其是涉及一種圖像聚類方法、裝置、電子設備、計算機可讀存儲介質。

背景技術

傳統的人臉聚類，是通過人臉識別技術對人臉圖片進行特征值解析，提取出圖片的相關要素信息，再通過比較特征值相似度，對圖片進行聚類。新的類簇與舊的類簇進行合并時，往往比較的是類簇質心的特征值相似度。如果滿足一定的閾值，則添加到類簇。

但是，采用單質心進行比對，會造成數據漂移的問題。

發明內容

本發明提供一種圖像聚類方法、裝置、電子設備、計算機可讀存儲介質，用以解決聚類過程中數據漂移問題。

為解決上述技術問題，本發明提供的第一個技術方案為：提供一種圖像聚類方法，包括：獲取第一類簇的多個質心和第二類簇的多個質心；對比第一類簇的多個質心與第二類簇的多個質心的相似度，根據相似度的比較結果，確定第二類簇與第一類簇能否合并；若第二類簇與第一類簇能合并，則將第二類簇與第一類簇進行合并，得到合并后的類簇的質心；利用合并后的類簇的質心進行圖像聚類。

其中，獲取第一類簇的多個質心和第二類簇的多個質心，包括：選取所述第一類簇和所述第二類簇中質量分數最高的圖像作為所述多個質心；和/或選取第一類簇和所述第二類簇中聚類時間最晚的圖像作為所述多個質心；和/或計算所述第一類簇中所有圖像對應的特征維度的第一平均值，根據所述第一類簇中每一圖像的特征維度與所述第一平均值的距離選取對應的圖像作為所述多個質心；及計算所述第二類簇中所有圖像對應的特征維度的第二平均值，根據所述第二類簇中每一圖像的特征維度與所述第二平均值的距離選取對應的圖像作為所述多個質心；和/或對所述第一類簇和所述第二類簇中所有圖像進行聚類，以得到所述多個質心。

其中，計算所述第一類簇中所有圖像對應的特征維度的第一平均值，根據所述第一類簇中每一圖像的特征維度與所述第一平均值的距離選取對應的圖像作為所述多個質心；及計算所述第二類簇中所有圖像對應的特征維度的第二平均值，根據所述第二類簇中每一圖像的特征維度與所述第二平均值的距離選取對應的圖像作為所述多個質心，包括：選取所述第一類簇中與所述第一平均值距離最近的特征維度對應的圖像作為所述多個質心；及選取所述第二類簇中與所述第二平均值距離最近的特征維度對應的圖像作為所述多個質心。

其中，所述對所述第一類簇和所述第二類簇中所有圖像進行聚類，以得到所述多個質心，包括：對所述第一類簇中所有圖像的特征維度進行降維處理，以得到部分所述第一圖像，及對所述第二類簇中所有圖像的特征維度進行降維處理，以得到部分所述第二圖像；對部分所述第一圖像進行聚類，以得到所述多個質心；及對部分所述第二圖像進行聚類，以得到所述多個質心。

其中，對比第一類簇的多個質心與第二類簇的多個質心的相似度，根據相似度的比較結果，確定第二類簇與第一類簇能否合并包括：利用第二類簇中所有質心對應的特征維度計算第二類簇中所有質心的平均值，作為第一虛擬質心；以及利用第一類簇中所有質心對應的特征維度計算第一類簇中所有質心的平均值，作為第二虛擬質心；計算第一虛擬質心與第二虛擬質心的距離值，若距離值小于閾值，則第二虛擬質心與第一虛擬質心相似度高，確定第二類簇與第一類簇能合并。