Attentive multi -stage convolutional neural network for crowd counting

被引:16
作者
Zhu, Ming [1 ]
Wang, Xuqing [1 ]
Tang, Jun [1 ]
Wang, Nian [1 ]
Qu, Lei [1 ]
机构
[1] Anhui Univ, Sch Elect & Informat Engn, Hefei 230601, Anhui, Peoples R China
基金
中国国家自然科学基金;
关键词
Convolutional neural network; Crowd counting; Density estimation; Soft attention mechanism;
D O I
10.1016/j.patrec.2020.05.009
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Crowd counting is an important problem in computer vision, whose application can be found in a wide range of tasks. Although this problem has been well studied, how to effectively deal with scale variations and perspective distortions is still a challenge. High-quality crowd density map depends heavily on how well these problems are solved. In this paper, we propose a novel network architecture called Attentive Multi-stage CNN for Crowd Counting (AMCNN). The AMCNN contains two subnetworks, i.e., hierarchical density estimator(HDE) and auxiliary count classifier (AUCC). The HDE adopts a hierarchical strategy to mine semantic features in a coarse-to-fine manner to tackle the problem of scale changes and perspective distortions. And the obtained composite features are used to generate the final density map. In addition, to further improve the density map quality, a soft attention mechanism is integrated into the AMCNN to distinct the foreground and the background. Furthermore, the AUCC is employed to achieve the count classification task, which is complementary to the task of density estimation. We evaluate our model on three public datasets: ShanghaiTech, UCF_CC_50 and Mall. Extensive experiments demonstrate that our counting model is on par with some state-of-the-art methods. Source code will be released at:https://github.com/wxq-ahu/crowd-count-amcnn. © 2020
引用
收藏
页码:279 / 285
页数:7
相关论文
共 40 条
[1]  
[Anonymous], IEEE INT C COMP VIS
[2]  
[Anonymous], 2017, PATTERN RECOGNIT LET
[3]  
[Anonymous], MULTIMED TOOLS APPL
[4]  
[Anonymous], 2018, IEEE C COMP VIS PATT
[5]  
[Anonymous], NEURAL COMPUT
[6]  
[Anonymous], 2017, IEEE C COMP VIS PATT
[7]  
[Anonymous], 2015, IEEE C COMP VIS PATT
[8]  
[Anonymous], 2016, IEEE T CIRCUITS SYST, DOI [10.1109/TCSVT.2016.2637379, DOI 10.1109/TCSVT.2016.2637379]
[9]  
[Anonymous], 2017, P IEEE C COMP VIS PA
[10]  
[Anonymous], 2016, P IEEE C COMP VIS PA