一种面向人群计数的卷积注意力网络模型

doi:10.3778/j.issn.1002-8331.2107-0131

摘要/Abstract

摘要： 随着当今国际社会形势逐渐复杂，公共安全和社会稳定面临严峻挑战。视频监控作为维护社会安定与建设智慧城市的重要手段，广泛应用于城市安全管理。高效的人群计数是实现基于视频进行安全管理的一个难点问题，旨在分析计算视频或图片场景中的人数。人群计数对控制关键场所人数、指挥公共交通、控制疫情蔓延、保障社会稳定具有重要积极意义。然而，人群计数问题仍然存在背景干扰、目标遮挡、目标尺度不一和目标分布不均等挑战，导致计数准确度较低。为了解决这些问题，梳理了人群计数发展的时间线，分析了现有方法的不足，并针对这些不足提出了基于相似性度量的卷积注意力网络。该方法结合基于相似性度量的损失函数和基于注意力机制的卷积神经网络模块，有效缓解了人群计数中背景干扰、目标遮挡、目标尺度不一和目标分布不均四个问题。通过在数据集上的实验和相关对比分析发现，基于相似性度量的卷积注意力网络具有很好的准确性和稳定性。

关键词: 人群计数, 卷积神经网络, 相似性度量, 注意力机制

Abstract: With the increasingly complex international situation, public security and social stability are facing severe challenges. As an important means to maintain social stability and develop smart cities, video surveillance has been widely used for managing public security. Crowd counting is a critical issue in video surveillance-based security management and it aims to calculate the number of people in the video or picture scene. It is of high significance for controlling the number of people in key places, directing public transportation, controlling the spread of the epidemic, and ensuring social stability. However, there are still some challenges in crowd counting, such as background interference, target occlusion, target scale difference and target distribution inequality. In order to solve these challenges, this paper reviews the timeline of crowd counting development, analyzes the shortcomings of existing methods, and puts forward corresponding solutions. Then, the paper proposes a convolutional attention network based on similarity measure, which combines the loss function based on similarity measure and the convolutional neural network module based on attention mechanism, and effectively alleviates the four issues of background interference, target occlusion, target scale difference and uneven distribution in crowd counting. Through the experiments and comparative analysis on datasets, it is found that the convolutional attention network based on similarity measure achieves good accuracy and stability in crowd counting.

Key words: crowd counting, convolutional neural network, similarity measure, attention mechanism

朱宇斌, 李文根, 关佶红, 张毅超. 一种面向人群计数的卷积注意力网络模型[J]. 计算机工程与应用, 2023, 59(1): 156-161.

ZHU Yubin, LI Wengen, GUAN Jihong, ZHANG Yichao. Convolutional Attention Network for Crowd Counting[J]. Computer Engineering and Applications, 2023, 59(1): 156-161.

参考文献

[1] SINDAGI V A，PATEL V M.A survey of recent advances in cnn-based single image crowd counting and density estimation[J].Pattern Recognition Letters，2018，107：3-16.
[2] WU B，NEVATIA R.Detection of multiple，partially occluded humans in a single image by bayesian combination of edgelet part detectors[C]//Proceedings of IEEE International Conference on Computer Vision，2005：90-97.
[3] BROSTOW G J，CIPOLLA R.Unsupervised Bayesian detection of independent motion in crowds[C]//Proceedings of IEEE Computer Society Conference on Computer Vision and Pattern Recognition，2006：594-601.
[4] RABAUD V，BELONGIE S.Counting crowded moving objects[C]//Proceedings of IEEE Computer Society Conference on Computer Vision and Pattern Recognition，2006：705-711.
[5] CHAN A B，LIANG Z S J，VASCONCELOS N.Privacy preserving crowd monitoring：Counting people without people models or tracking[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition，2008：1-7.
[6] CHO S Y，CHOW T W，LEUNG C T.A neural-based crowd estimation by hybrid global learning algorithm[J].IEEE Transactions on Systems，Man，and Cybernetics（Part B），1999，29（4）：535-541.
[7] CHEN K，GONG S，XIANG T，et al.Cumulative attribute space for age and crowd density estimation[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition，2013：2467-2474.
[8] 张友梅.基于注意力卷积神经网络的人群计数算法研究[D].济南：山东大学，2019.
ZHANG Y M.Research on crowd counting algorithm based on attention convolutional neural network[D].Jinan：Shandong University，2019.
[9] ZHANG Y，ZHOU D，CHEN S，et al.Single-image crowd counting via multi-column convolutional neural network[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition，2016：589-597.
[10] SAM D B，SURYA S，BABU R V.Switching convolutional neural network for crowd counting[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition，2017：4031-4039.
[11] MA Z，WEI X，HONG X，et al.Bayesian loss for crowd count estimation with point supervision[C]//Proceedings of IEEE International Conference on Computer Vision，2019：6142-6151.
[12] SIMONYAN K，ZISSERMAN A.Very deep convolutional networks for large-scale image recognition[J].arXiv：1409.1556，2014.
[13] 朱宇斌.基于相似性度量的人群计数方法[J].电脑知识与技术，2021，16（6）：179-180.
ZHU Y B.Crowd counting method based on similarity measurement[J].Computer Knowledge and Technology，2021，16（6）：179-180.
[14] SHI Z，ZHANG L，SUN Y，et al.Multiscale multitask deep NetVLAD for crowd counting[J].IEEE Transactions on Industrial Informatics，2018，14（11）：4953-4962.
[15] WANG Q，GAO J，LIN W，et al.NWPU-Crowd：A large-scale benchmark for crowd counting and localization[J].arXiv：2001.03360，2020.
[16] ZHOU D，HE Q.Cascaded multi-task learning of head segmentation and density regression for RGBD crowd counting[J].IEEE Access，2020，8：101616-101627.
[17] SHEN Z，XU Y，NI B，et al.Crowd counting via adversarial cross-scale consistency pursuit[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition，2018：5245-5254.
[18] LI Y，ZHANG X，CHEN D.CsrNet：Dilated convolutional neural networks for understanding the highly congested scenes[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition，2018：1091-1100.
[19] SHI Z，METTES P，SNOEK C G.Counting with focus for free[C]//Proceedings of IEEE International Conference on Computer Vision，2019：4200-4209.
[20] SINDAGI V A，PATEL V M.HA-CCN：Hierarchical attention-based crowd counting network[J].IEEE Transactions on Image Processing，2019，29：323-335.
[21] JIANG X，XIAO Z，ZHANG B，et al.Crowd counting and density estimation by trellis encoder-decoder networks[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition，2019：6133-6142.
[22] LIU W，SALZMANN M，FUA P.Context-aware crowd counting[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition，2019：5099-5108.
[23] WANG Q，GAO J，LIN W，et al.Learning from synthetic data for crowd counting in the wild[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition，2019：8198-8207.