摘要
人群计数是计算机视觉领域的重要任务。交通场景中的人群计数任务对于维护公众出行安全、实现交通智能化具有重要作用。公共交通场景中通常存在行人相互遮挡、背景复杂等现象,给人群计数带来了困难。为了实现高精度的人群计数,研究了基于注意力机制的人群密度估计网络。网络包含3个部分:特征提取模块通过生成多尺度的特征图,增强网络的特征表达能力,提高网络对行人大小变化的鲁棒性;注意力模块通过抑制背景噪声响应,强化人群特征响应,生成特征图中人群区域的概率分布,增强网络区分人群区域与背景区域的能力;密度估计模块在注意力机制的约束下指导网络回归高分辨率的人群密度图,提高网络对人群区域的敏感性。设计了基于背景感知的结构损失函数,能够降低模型的错误识别率,提高模型的计数准确率;采用多级监督机制指导网络进行学习,能够帮助梯度反向传播和减少过度拟合,进一步提高网络的人群计数精度。在公共数据集ShanghaiTech上进行了实验,实验结果表明:与目前最先进的算法相比,在ShanghaiTechA和ShanghaiTechB数据集上,平均绝对误差(mean absolute error,MAE)分别提高了2.4%和1.5%,均方误差(mean square error,MSE)分别提高了3.3%和0.9%,证明了提出的算法在人群拥挤和稀疏的场景中均有更好的准确性和鲁棒性。同时,在真实场景数据集上进行了实验,MAE=7.7,MSE=12.6,证明了提出的算法具有良好的实用性。
Crowd count is an important task in computer vision.Crowd count task in traffic scenes plays a signifi-cant role in maintaining public traffic safety and achieving traffic intelligence.However,crowd count in public traf-fic scenes faces difficulties due to pedestrian occlusion and complex background.In order to achieve high accuracy crowd count,an attention-based crowd density estimation network is proposed.The network consists of three parts:a feature extraction module is designed to generate multi-scale feature maps,which can enhance the feature repre-sentation capability and improve the robustness to pedestrian scale variation of the network;an attention module is designed to suppress the background noise response and enhance the crowd feature response,generate the probabili-ty distribution of the crowd region in the feature map,which can enhance the ability of the network to distinguish the crowd region from the background region;a density estimation module is designed that guides the network to re-gress a high-resolution crowd density map under the constraint of attention mechanism,which can improve the sen-sitivity of the network to crowd regions.In addition,a background-aware structure loss function is designed to re-duce the model false recognition rate and improve the model counting accuracy;meanwhile,a multi-level super-vi-sion mechanism is adopted to guide the network for learning,which can help gradient back-propagation and reduce over-fitting,further improving the network's crowd count accuracy.Experiments are carried out on public dataset ShanghaiTech.Compared with the state-of-the-art algorithms,on ShanghaiTechA and ShanghaiTechB datasets,the mean absolute error(MAE)improves by 2.4%and 1.5%,and the mean square error(MSE)improves by 3.3%and 0.9%,respectively,which demonstrates the superior accuracy and robustness of the proposed algorithm in both crowded and sparse scenes.Experiments are also conducted on real scene dataset with MAE=7.7 and MSE=12.6,which proves the good applicability of the proposed algorithm.
作者
王丽园
姚韵涛
贾洋
肖进胜
李必军
WANG Liyuan;YAO Yuntao;JIA Yang;XIAO Jinsheng;LI Bijun(CCCC Second Highway Consultants Co.,LTD,Wuhan 430056,China;School of Electronic Information,Wuhan University,Wuhan 430072,China;Sichuan Highway Planning,Survey,Design and Research Institute Co.,LTD.,Chengdu 610041,China;State Key Laboratory of Information Engineering in Surveying,mapping and Remote Sensing,Wuhan University,Wuhan 430079,China)
出处
《交通信息与安全》
CSCD
北大核心
2023年第6期107-113,共7页
Journal of Transport Information and Safety
基金
湖北省重点研发计划项目(2023BAB022)
中国交通建设集团有限公司科技研发项目(编号2019-ZJKJ-ZDZX02)。
关键词
交通安全
人群计数
注意力机制
背景感知结构损失
多级监督机制
traffic safety
crowd count
attention mechanism
background-aware structure loss algorithm
multi-lev-el supervision
