[發明專利]一種行人重識別方法、裝置和計算機設備

有效

• 申請號: 201910320299.1
• 申請日: 2019-04-19
• 公開（公告）號: CN111832361B
• 公開（公告）日: 2023-08-29
• 發明（設計）人: 張鼎
• 申請（專利權）人: 杭州海康威視數字技術股份有限公司
• 主分類號: G06V40/10
• 分類號: G06V40/10;G06V10/74;G06V10/46;G06V10/82;G06F16/583
• 代理公司: 北京博思佳知識產權代理有限公司 11415
• 代理人: 林祥

1.一種行人重識別方法，其特征在于，所述方法包括：

確定待查詢行人圖片的類別信息；其中，所述類別信息用于表征所述待查詢行人圖片包含的人體部件所屬的類別；

從預先建立的各指定人體部件對應的比對庫中選取與所述類別信息匹配的比對庫作為目標比對庫；

計算所述待查詢行人圖片與所述目標比對庫中的目標數據之間的相似度，并依據計算得到的相似度，對所述待查詢行人圖片進行行人重識別。

2.根據權利要求1所述的方法，其特征在于，所述確定待查詢行人圖片的類別信息，包括：

將所述待查詢行人圖片輸入到預先訓練好的人體部件分類網絡，由所述人體部件分類網絡對所述待查詢行人圖片進行特征提取和分類后，輸出所述待查詢行人圖片屬于各所述指定人體部件的概率，并將最大概率對應的指定人體部件確定為所述待查詢行人圖片所屬的類別；

將所述待查詢行人圖片所屬的類別確定為所述類別信息。

3.根據權利要求1所述的方法，其特征在于，所述目標數據為人體部件圖片對應的特征圖，所述計算所述待查詢行人圖片與所述目標比對特征庫中的目標數據之間的相似度，包括：

對所述待查詢行人圖片進行特征提取，得到所述待查詢行人圖片對應的目標特征圖；

計算所述目標特征圖與所述目標比對特征庫中的各人體部件圖片對應的特征圖之間的相似度。

4.根據權利要求3所述的方法，其特征在于，所述各指定人體部件對應的比對庫采用如下方法建立：

從預先采集的行人圖片中截取包含各所述指定人體部件的人體部件圖片；

對各所述人體部件圖片進行特征提取，得到各所述人體部件圖片對應的特征圖；

按照各所述人體部件圖片所包含的人體部件，將各所述人體部件圖片對應的特征圖分類存儲，得到各所述指定人體部件對應的比對庫。

5.根據權利要求3或4所述的方法，其特征在于，對圖片進行特征提取的步驟，包括：

將該圖片輸入到預先訓練好的特征提取網絡，由所述特征提取網絡對該圖片進行特征提取后輸出該圖片對應的特征圖。

6.根據權利要求1所述的方法，其特征在于，所述目標數據為人體部件圖片；所述計算所述待查詢行人圖片與所述目標比對特征庫中的目標數據之間的相似度，包括：

對所述待查詢行人圖片進行特征提取，得到所述待查詢行人圖片對應的目標特征圖；

對所述目標比對庫中的各人體部件圖片進行特征提取，得到所述目標比對庫中的各人體部件圖片對應的特征圖；

計算所述目標特征圖與所述目標比對特征庫中的各人體部件圖片對應的特征圖之間的相似度。

7.一種行人重識別裝置，其特征在于，所述裝置包括分類模塊、選擇模塊和處理模塊，其中，

所述分類模塊，用于確定待查詢行人圖片的類別信息；其中，所述類別信息用于表征所述待查詢行人圖片包含的人體部件所屬的類別；

所述選擇模塊，用于從預先建立的各指定人體部件對應的比對庫中選取與所述類別信息匹配的比對庫作為目標比對庫；

所述處理模塊，用于計算所述待查詢行人圖片與所述目標比對庫中的目標數據之間的相似度，并依據計算得到的相似度，對所述待查詢行人圖片進行行人重識別。

8.根據權利要求7所述的裝置，其特征在于，所述分類模塊，具體用于將將所述待查詢行人圖片輸入到預先訓練好的人體部件分類網絡，由所述人體部件分類網絡對所述待查詢行人圖片進行特征提取和分類后，輸出所述待查詢行人圖片屬于各所述指定人體部件的概率，并將最大概率對應的指定人體部件確定為所述待查詢行人圖片所屬的類別；將所述待查詢行人圖片所屬的類別確定為所述類別信息。

9.一種計算機可讀存儲介質，其上存儲有計算機程序，其特征在于，所述程序被處理器執行時實現權利要求1-6任一項所述方法的步驟。

10.一種計算機設備，包括存儲器、處理器及存儲在存儲器上并可在處理器上運行的計算機程序，其特征在于，所述處理器執行所述程序時實現權利要求1-6任一項所述方法的步驟。