[發明專利]一種人群計數方法、裝置、計算機設備及存儲介質

有效

• 申請號: 202110191656.6
• 申請日: 2021-02-19
• 公開（公告）號: CN112991274B
• 公開（公告）日: 2023-06-30
• 發明（設計）人: 劉釗
• 申請（專利權）人: 平安科技（深圳）有限公司
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06T5/50;G06T3/40;G06N3/0464;G06V20/52;G06V10/82
• 代理公司: 深圳市世聯合知識產權代理有限公司 44385
• 代理人: 汪琳琳

1.一種人群計數方法，其特征在于，包括下述步驟：

構建人群計數通用模型，所述人群計數通用模型包括金字塔池化模塊與卷積模塊，其中所述金字塔池化模塊包括多層金字塔層；所述每層金字塔層包括池化層、第一卷積層以及上采樣層；所述卷積模塊包括第二卷積層和第三卷積層；

將多個原始圖片特征輸入到所述金字塔池化模塊中，根據每層金字塔層預設的輸出特征尺寸進行不同尺度的池化，得到第一多尺度特征圖；

將所述第一多尺度特征圖輸入卷積模塊進行卷積操作輸出第二多尺度特征圖；

將所述第二多尺度特征圖與原始圖片特征進行拼接融合得到第三多尺度特征圖；及

將所述第三多尺度特征圖進行解碼后轉化為人群密度圖；

其中，所述將多個原始圖片特征輸入到所述金字塔池化模塊中，根據每層金字塔層預設的輸出特征尺寸進行不同尺度的池化，得到第一多尺度特征圖的步驟包括：

將所述原始圖片特征分別輸入至每層金字塔層的池化層中進行池化運算，在每層所述金字塔層上得到對應的第一特征圖；

將所述第一特征圖經過所述第一卷積層進行第一卷積操作，輸出對應的第一卷積特征圖；

對所述第一卷積特征圖輸入到上采樣層進行上采樣操作，輸出預設大小的第一尺度特征圖；

將每層的所述第一尺度特征圖在通道維度上進行拼接得到第一多尺度特征圖；

其中，所述將所述第一多尺度特征圖輸入卷積模塊進行卷積操作輸出第二多尺度特征圖的步驟包括：

將所述第一多尺度特征圖輸入到第二卷積層進行第二卷積操作，得到第二卷積特征圖；

調整所述第二卷積層的輸出通道數并輸出所述第二卷積特征圖；

把所述第二卷積特征圖輸入到第三卷積層進行第三卷積操作并輸出第二多尺度特征圖。

2.根據權利要求1所述的人群計數方法，其特征在于，所述將所述第二多尺度特征圖與原始圖片特征進行拼接融合得到第三多尺度特征圖的步驟包括：

將所述第二多尺度特征圖與原始圖片特征按照通道維度進行拼接得到第三拼接特征；

使用1*1卷積核對所述第三拼接特征進行融合得到第三多尺度特征圖。

3.根據權利要求1所述的人群計數方法，其特征在于，所述將所述第三多尺度特征圖進行解碼后轉化為人群密度圖的步驟包括：

使用多層卷積層對所述第三多尺度特征圖進行解碼；

將解碼后的第三多尺度特征圖的空間尺寸恢復至原始圖片尺寸，得到人群密度圖。

4.根據權利要求3所述的人群計數方法，其特征在于，所述將解碼后的第三多尺度特征圖的空間尺寸恢復至原始圖片尺寸得到人群密度圖的步驟包括：

采用雙線性插值法對解碼后的第三多尺度特征圖進行上采樣，得到與原始圖片等大的尺寸。

5.根據權利要求1至4中任一項所述的人群計數方法，其特征在于，在所述將所述第三多尺度特征進行解碼后轉化為人群密度圖的步驟之后還包括：

對所述人群密度圖中每個像素點的值求積分得到人群密度估計，將所有像素點的值相加求和，得到總人數計數。