arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2104.02245cs.CVcs.AI
收藏

基于注意力引导的多尺度上下文聚合网络的人群计数

Multi-Scale Context Aggregation Network with Attention-Guided for Crowd Counting

Beijing Jiaotong University(北京交通大学);Shenzhen University(深圳大学)

Xin Wang, Yang Zhao, Tangwen Yang, Qiuqi Ruan

摘要:

人群计数的目的是预测人群的数量,并生成图像中的密度图。有许多挑战,包括不同的头部比例,人群分布在图像和混乱的背景的多样性。在本文中,我们提出了一个多尺度上下文聚合网络(MSCANet)的基础上,单列编码器-解码器架构的人群计数,它包括一个编码器的基础上密集的上下文感知模块(DCAM)和分层注意力引导解码器。为了处理尺度变化的问题,我们构建了DCAM聚合多尺度的上下文信息,通过密集连接的扩张卷积与不同的感受野。由于DCAM具有远距离感受野和密集尺度采样的特点,因此能够捕捉到人群区域丰富的上下文信息。此外,为了抑制背景噪声和生成高质量的密度图,我们在解码器中采用了分层注意力引导机制。这有助于通过引入基于语义注意模块(SAM)的多重监督,从编码器的浅层特征图中集成更多有用的空间信息。大量的实验表明,该方法实现了更好的性能比其他类似的国家的最先进的方法在三个具有挑战性的基准数据集的人群计数。代码可在https://github.com/KingMV/MSCANet上获得

英文摘要:

Crowd counting aims to predict the number of people and generate the density map in the image. There are many challenges, including varying head scales, the diversity of crowd distribution across images and cluttered backgrounds. In this paper, we propose a multi-scale context aggregation network (MSCANet) based on single-column encoder-decoder architecture for crowd counting, which consists of an encoder based on a dense context-aware module (DCAM) and a hierarchical attention-guided decoder. To handle the issue of scale variation, we construct the DCAM to aggregate multi-scale contextual information by densely connecting the dilated convolution with varying receptive fields. The proposed DCAM can capture rich contextual information of crowd areas due to its long-range receptive fields and dense scale sampling. Moreover, to suppress the background noise and generate a high-quality density map, we adopt a hierarchical attention-guided mechanism in the decoder. This helps to integrate more useful spatial information from shallow feature maps of the encoder by introducing multiple supervision based on semantic attention module (SAM). Extensive experiments demonstrate that the proposed approach achieves better performance than other similar state-of-the-art methods on three challenging benchmark datasets for crowd counting. The code is available at https://github.com/KingMV/MSCANet

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑