4 个月前

用于人群计数的切换卷积神经网络

用于人群计数的切换卷积神经网络

摘要

我们提出了一种新颖的人群计数模型,该模型可以将给定的人群场景映射到其密度。人群分析受到多种因素的影响,如由于极端拥挤导致的人与人之间的遮挡、人与背景元素的高度相似性以及摄像机视角的较大变化。目前最先进的方法通过使用多尺度卷积神经网络(CNN)架构、递归网络以及具有不同感受野的多列CNN特征的后期融合来应对这些因素。我们提出了一种切换卷积神经网络,该网络利用图像内人群密度的变化来提高预测的人群计数的准确性和定位精度。在训练过程中,根据每个CNN对人群计数预测的质量,将来自人群场景网格中的图像块传递给独立的CNN回归器。这些独立的CNN回归器被设计为具有不同的感受野,并且训练了一个切换分类器以将人群场景图像块传递给最佳的CNN回归器。我们在所有主要的人群计数数据集上进行了广泛的实验,并证明了相比当前最先进的方法有更好的性能。我们提供了从切换机制推断出的人群场景图像块空间多分法的可解释表示。观察结果表明,切换机制会根据人群密度将图像块传递给特定的CNN列。

代码仓库

基准测试

基准方法指标
crowd-counting-on-shanghaitech-aSwitch-CNN
MAE: 90.4
crowd-counting-on-shanghaitech-bSwitch-CNN
MAE: 21.6
crowd-counting-on-ucf-cc-50Switch-CNN
MAE: 318.1
crowd-counting-on-ucf-qnrfSwitch-CNN
MAE: 228
crowd-counting-on-veniceSwitch-CNN
MAE: 52.8
crowd-counting-on-worldexpo10Switch-CNN
Average MAE: 9.4

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
用于人群计数的切换卷积神经网络 | 论文 | HyperAI超神经