基于注意力机制的人群计数方法

为准确预测固定场景中的人群计数,在人群分析领域,采用一种融合注意力机制的卷积神经网络(CNN)进行人群计数,该模块结合空间域注意力和通道域注意力,空间域注意力可以编码整个图像的像素级上下文信息,以更准确地表达像素级别的密度图,而通道域注意力可以在不同的通道中提取更多的区分特征使网络显著表达人群的局部区域,并在多个公开数据集上进行测试。结果表明:基于注意力机制的人群计数方法可以准确地估计拥挤场景中的人群数量,在平均完全误差和均方误差上均优于CSRNet。

Method of crowd counting based on attention mechanism

In order to accurately predict crowd count in a fixed scene,in the field of crowd analysis,a convolutional neural network(CNN)integrating attention mechanism was adopted,which combined spatial domain attention and channel domain attention.The former could encode pixel-level context information of the entire image to express pixel-level density map more accurately,while the latter could extract more distinguishing features in different channels to significantly express local area of the crowd.Through tests on multiple public data sets,it is found that the crowd counting method based on attention mechanism can accurately estimate number of people in crowded scenes,and it proves better than CSRNet in terms of mean absolute error and mean square error.