基于多重注意力引导的人群计数算法

摘要

针对实际场景中存在的人群非均匀分布问题,提出了一种基于多重注意力引导的人群计数算法。首先,基于轻量级金字塔切分注意力机制构建了自顶向下的特征融合路径,旨在促进高层语义信息和低层空间细节的融合,生成高级语义和空间细节兼备的高质量特征图;然后,提取并融合多尺度上下文信息,以此生成关注于不同密度分布模式的注意力权重图;最后,通过注意力权重图指导密度回归网络识别不同分布状态下的行人目标,增强模型对密度变化的适应性,生成高质量人群密度图。在ShanghaiTech、UCF_QNRF和JHU-CROWD++三个数据集上进行了大量的实验来说明所提算法的先进性。