基于尺度空间金字塔网络的人群计数算法

摘要

针对人群密度大和分布不均等因素造成计数困难的问题,提出了一种基于尺度空间金字塔网络的人群计数算法。首先通过迁移经典网络VGG16的前10层作为前端模块提取初步特征;然后使用尺度空间金字塔模块提取多尺度人群特征,并在该模块中引入扩张卷积,通过增大感受野进一步优化特征图,以解决人群拥挤和人群分布不均的问题;最后使用1×1的卷积层将特征图回归为密度图。实验结果表明,与对比算法相比,所提算法的平均绝对误差(mean absolute error,MAE)和均方误差(mean square error,MSE)均显著降低,可取得较好的人群计数效果。