摘要
本发明公开基于注意力机制的单张图像人群计数方法。所述方法包括对当前图像进行逐通道像素Z-Score归一化;设计4+2+2网络提取图像纹理特征;设计编-解码网络提取图像全局注意力特征;逐通道将纹理特征与全局注意力特征进行元素乘,生成融合特征;对融合特征进行1*1卷积运算生成人群密度图,对人群密度图进行积分获得计数结果。本方法设计的网络能够端到端地进行训练和测试,能够提高行人非均匀分布场景下的人群计数精度,在World Expo10数据集的5个测试场景中平均计数误差不大于3.28人。
- 单位