[發明專利]客流統計方法、裝置、電子設備和存儲介質

有效

• 申請號: 202011326554.2
• 申請日: 2020-11-24
• 公開（公告）號: CN112132119B
• 公開（公告）日: 2021-05-28
• 發明（設計）人: 李小兵;胡克任;茆曉萱;王旭;支洪平;劉德貴
• 申請（專利權）人: 科大訊飛（蘇州）科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62;G06T7/246
• 代理公司: 北京路浩知識產權代理有限公司 11002
• 代理人: 程琛

1.一種客流統計方法，其特征在于，包括：

對待統計視頻中的當前視頻幀進行頭肩檢測，得到所述當前視頻幀中的若干個頭肩區域圖像，所述頭肩區域圖像是從所述當前視頻幀中截取的包含有單個行人頭肩的圖像，所述待統計視頻是基于垂直視角采集的；

將各個頭肩區域圖像輸入至頭肩特征提取模型，得到所述頭肩特征提取模型輸出的各個頭肩區域圖像的頭肩特征；其中，所述頭肩特征提取模型是基于樣本圖像對訓練得到的，所述樣本圖像對包括錨點頭肩圖像、與所述錨點頭肩圖像屬于相同行人的正樣本頭肩圖像，以及與所述錨點頭肩圖像屬于不同行人的負樣本頭肩圖像；

基于各個頭肩區域圖像的頭肩特征，更新所述待統計視頻的客流軌跡，基于所述客流軌跡進行客流統計；

其中，所述基于各個頭肩區域圖像的頭肩特征，更新所述待統計視頻的客流軌跡，包括：

將各個頭肩區域圖像的頭肩特征，與上次更新的客流軌跡中各個行人運動軌跡中各個歷史頭肩特征進行匹配，將匹配成功的頭肩區域圖像的頭肩特征添加至對應行人運動軌跡中；

所述將各個頭肩區域圖像的頭肩特征，與上次更新的客流軌跡中各個行人運動軌跡中各個歷史頭肩特征進行匹配，包括：

基于任一頭肩區域圖像的頭肩特征與任一行人運動軌跡中各個歷史頭肩特征之間的距離，以及各個歷史頭肩特征對應的距離權重，確定所述任一頭肩區域圖像的頭肩特征與所述任一行人運動軌跡之間的距離；其中，頭肩特征之間的距離為頭肩特征之間的相似度；

基于各個頭肩區域圖像的頭肩特征與各個行人運動軌跡之間的距離，構建代價矩陣；

基于所述代價矩陣，對各個頭肩區域圖像的頭肩特征與各個行人運動軌跡進行匹配。

2.根據權利要求1所述的客流統計方法，其特征在于，所述頭肩特征提取模型是以最大化所述錨點頭肩圖像與所述正樣本頭肩圖像的頭肩特征相似度，以及最小化所述錨點頭肩圖像與所述負樣本頭肩圖像的頭肩特征相似度為目標訓練得到的。

3.根據權利要求1所述的客流統計方法，其特征在于，所述頭肩特征提取模型包括特征融合層，以及按照尺度從大到小的順序排列的多個不同尺度的特征提取層；

所述將各個頭肩區域圖像輸入至頭肩特征提取模型，得到所述頭肩特征提取模型輸出的各個頭肩區域圖像的頭肩特征，包括：

將上一尺度的圖像特征輸入至當前尺度的特征提取層，得到當前尺度的特征提取層輸出的當前尺度的圖像特征；

將每個尺度的圖像特征輸入至所述特征融合層，得到所述特征融合層輸出的頭肩特征；

其中，首個尺度的特征提取層的輸入為任一頭肩區域圖像。

4.根據權利要求1所述的客流統計方法，其特征在于，所述對待統計視頻中的當前視頻幀進行頭肩檢測，得到所述當前視頻幀中的若干個頭肩區域圖像，包括：

對當前視頻幀進行頭肩檢測，得到當前視頻幀中的若干個候選頭肩檢出框；

基于所述待統計視頻中當前視頻幀的前一視頻幀，以及當前視頻幀，確定所述當前視頻幀的差分圖像；

對所述差分圖像進行頭肩檢測，得到當前視頻幀中的若干個運動頭肩檢出框；

基于每個候選頭肩檢出框和每個運動頭肩檢出框，確定所述當前視頻幀中的若干個頭肩區域圖像。

5.根據權利要求4所述的客流統計方法，其特征在于，所述基于每個候選頭肩檢出框和每個運動頭肩檢出框，確定所述當前視頻幀中的若干個頭肩區域圖像，包括：

若任一運動頭肩檢出框在任一候選頭肩檢出框中的面積占比大于等于預設占比閾值，則確定所述任一候選頭肩檢出框在所述當前視頻幀中的區域圖像為所述頭肩區域圖像。

6.根據權利要求1所述的客流統計方法，其特征在于，所述將各個頭肩區域圖像的頭肩特征，與上次更新的客流軌跡中各個行人運動軌跡中各個歷史頭肩特征進行匹配，包括：

基于任一行人運動軌跡中的各個歷史頭肩特征，確定所述任一行人運動軌跡的預估頭肩特征；

將各個頭肩區域圖像中與所述預估頭肩特征之間的距離小于預設距離閾值的頭肩特征，與所述任一行人運動軌跡中的各個歷史頭肩特征進行匹配；

若所述任一行人運動軌跡未匹配到頭肩特征，則將所述預估頭肩特征添加至所述任一行人運動軌跡。