[發明專利]模型訓練和人流量統計方法、裝置及設備

有效

• 申請號: 202211575979.6
• 申請日: 2022-12-09
• 公開（公告）號: CN115661586B
• 公開（公告）日: 2023-04-18
• 發明（設計）人: 白帆
• 申請（專利權）人: 云粒智慧科技有限公司
• 主分類號: G06V10/774
• 分類號: G06V10/774;G06V20/40;G06T7/73;G06T7/66;G06T7/246
• 代理公司: 北京路浩知識產權代理有限公司 11002
• 代理人: 張曉霞

本申請提供一種模型訓練和人流量統計方法、裝置及設備，涉及人工智能技術領域，該模型訓練方法包括：獲取訓練樣本，所述訓練樣本中包括多張樣本圖像、以及所述樣本圖像的標注信息，所述標注信息包括目標對象的標注位置和標注標識，所述多張樣本圖像為樣本視頻中連續的多張圖像；將所述多張樣本圖像輸入至對象識別模型，得到所述對象識別模型輸出的識別信息，所述識別信息包括目標對象的識別位置、識別標識和對應的樣本圖像；根據所述標注信息和所述識別信息，對所述對象識別模型的參數進行調整。減小了由于環境發生變化帶來的目標對象識別的不利影響，從而提高人流量統計的準確率和效果。

技術領域

本申請涉及人工智能技術領域，尤其涉及一種模型訓練和人流量統計方法、裝置及設備。

背景技術

人流量統計指的是通過一定的技術手段獲取實時的人流數據的過程，能夠對景區、商場等公共場所的精細化管理提供重要支撐。

目前，主要是采用視頻監控進行實時判斷人流量。具體的，通過對監控設備的重復利用，根據監控視頻中空間和外觀相似度進行匹配，來識別監控視頻中的人，從而進行人流量的統計。由于這種人流量統計方式是基于空間和外觀相似度匹配實現的，在環境發生變化（例如產生遮擋）時，目前的人流量統計的效果不佳。

發明內容

本申請涉提供一種模型訓練和人流量統計方法、裝置及設備，以提高目前人流量統計的效果。

第一方面，本申請提供一種模型訓練方法，包括：

獲取訓練樣本，所述訓練樣本中包括多張樣本圖像、以及所述樣本圖像的標注信息，所述標注信息包括目標對象的標注位置和標注標識，所述多張樣本圖像為樣本視頻中連續的多張圖像；

將所述多張樣本圖像輸入至對象識別模型，得到所述對象識別模型輸出的識別信息，所述識別信息包括目標對象的識別位置、識別標識和對應的樣本圖像；

根據所述標注信息和所述識別信息，對所述對象識別模型的參數進行調整。

在一種可能的實施方式中，所述對象識別模型包括主干網絡模塊、目標檢測模塊和全局跟蹤模塊；所述將所述多張樣本圖像輸入至對象識別模型，得到所述對象識別模型輸出的識別信息，包括：

基于所述主干網絡模塊對所述多張樣本圖像進行特征提取處理，得到所述多張樣本圖像的特征圖；

根據所述目標檢測模塊對所述特征圖進行目標檢測處理，得到所述多張樣本圖像對應的檢測數據；

基于所述全局跟蹤模塊對所述檢測數據和所述特征圖進行處理，得到所述識別信息。

在一種可能的實施方式中，所述目標檢測模塊包括第一檢測頭、第二檢測頭和第三檢測頭；所述根據所述目標檢測模塊對所述特征圖進行目標檢測處理，得到所述多張樣本圖像對應的檢測數據，包括：

根據所述第一檢測頭對所述特征圖進行中心點檢測處理，得到各所述樣本圖像中目標對象的中心點位置；

根據所述第二檢測頭對所述特征圖進行中心點偏移量檢測處理，得到各所述樣本圖像中目標對象的中心點偏移量；

根據所述第三檢測頭對所述特征圖進行錨點檢測處理，得到各所述樣本圖像中所述目標對象的中心點位置對應的錨點邊界框的尺寸；

所述檢測數據包括所述中心點位置、所述中心點偏移量和所述錨點邊界框的尺寸。

在一種可能的實施方式中，所述基于所述全局跟蹤模塊對所述檢測數據和所述特征圖進行處理，得到所述識別信息，包括：

根據所述檢測數據和所述特征圖，獲取所述多張樣本圖像中各所述目標對象的表征特征；

根據所述全局跟蹤模塊對所述表征特征進行全局跟蹤處理，得到所述識別信息。