[发明专利]一种人群计数方法、装置、计算机设备及存储介质

权利要求书

1.一种人群计数方法，其特征在于，包括下述步骤：

构建人群计数通用模型，所述人群计数通用模型包括金字塔池化模块与卷积模块，其中所述金字塔池化模块包括多层金字塔层；所述每层金字塔层包括池化层、第一卷积层以及上采样层；所述卷积模块包括第二卷积层和第三卷积层；

将多个原始图片特征输入到所述金字塔池化模块中，根据每层金字塔层预设的输出特征尺寸进行不同尺度的池化，得到第一多尺度特征图；

将所述第一多尺度特征图输入卷积模块进行卷积操作输出第二多尺度特征图；

将所述第二多尺度特征图与原始图片特征进行拼接融合得到第三多尺度特征图；及

将所述第三多尺度特征图进行解码后转化为人群密度图；

其中，所述将多个原始图片特征输入到所述金字塔池化模块中，根据每层金字塔层预设的输出特征尺寸进行不同尺度的池化，得到第一多尺度特征图的步骤包括：

将所述原始图片特征分别输入至每层金字塔层的池化层中进行池化运算，在每层所述金字塔层上得到对应的第一特征图；

将所述第一特征图经过所述第一卷积层进行第一卷积操作，输出对应的第一卷积特征图；

对所述第一卷积特征图输入到上采样层进行上采样操作，输出预设大小的第一尺度特征图；

将每层的所述第一尺度特征图在通道维度上进行拼接得到第一多尺度特征图；

其中，所述将所述第一多尺度特征图输入卷积模块进行卷积操作输出第二多尺度特征图的步骤包括：

将所述第一多尺度特征图输入到第二卷积层进行第二卷积操作，得到第二卷积特征图；

调整所述第二卷积层的输出通道数并输出所述第二卷积特征图；

把所述第二卷积特征图输入到第三卷积层进行第三卷积操作并输出第二多尺度特征图。

2.根据权利要求1所述的人群计数方法，其特征在于，所述将所述第二多尺度特征图与原始图片特征进行拼接融合得到第三多尺度特征图的步骤包括：

将所述第二多尺度特征图与原始图片特征按照通道维度进行拼接得到第三拼接特征；

使用1*1卷积核对所述第三拼接特征进行融合得到第三多尺度特征图。

3.根据权利要求1所述的人群计数方法，其特征在于，所述将所述第三多尺度特征图进行解码后转化为人群密度图的步骤包括：

使用多层卷积层对所述第三多尺度特征图进行解码；

将解码后的第三多尺度特征图的空间尺寸恢复至原始图片尺寸，得到人群密度图。

4.根据权利要求3所述的人群计数方法，其特征在于，所述将解码后的第三多尺度特征图的空间尺寸恢复至原始图片尺寸得到人群密度图的步骤包括：

采用双线性插值法对解码后的第三多尺度特征图进行上采样，得到与原始图片等大的尺寸。

5.根据权利要求1至4中任一项所述的人群计数方法，其特征在于，在所述将所述第三多尺度特征进行解码后转化为人群密度图的步骤之后还包括：

对所述人群密度图中每个像素点的值求积分得到人群密度估计，将所有像素点的值相加求和，得到总人数计数。