arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2007.14245cs.CVcs.LGstat.ML

贝叶斯多尺度神经网络用于人群计数

Bayesian Multi-Scale Neural Network for Crowd Counting

University of Maryland, College Park(马里兰大学 College Park 分校)

Abhinav Sagar

AI总结:

提出一种集成ResNet、扩张卷积和透视感知聚合模块的贝叶斯多尺度神经网络,用于人群计数,在三个基准数据集上超越现有方法,并提供不确定性估计。

摘要:

人群计数是计算机视觉中一项具有挑战性且至关重要的任务,其应用范围从公共安全到城市规划。近年来,使用卷积神经网络(CNN)估计密度图的方法取得了显著成功。然而,由于严重的遮挡、尺度变化和透视畸变,在高度拥挤的场景中准确计数个体仍然是一个未解决的问题,在这些场景中,人们以截然不同的尺寸出现在图像中。在这项工作中,我们提出了一种新颖的深度学习架构,有效应对这些挑战。我们的网络集成了基于ResNet的特征提取器,用于捕获丰富的层次表示,随后是一个采用扩张卷积的下采样块,以在扩大感受野的同时保持空间分辨率。一个使用转置卷积的上采样块重建高分辨率密度图。我们架构的核心是一个新颖的透视感知聚合模块(PAM),旨在通过自适应聚合多尺度上下文信息来增强对尺度和透视变化的鲁棒性。我们详细介绍了训练过程,包括使用的损失函数和优化策略。我们的方法在三个广泛使用的基准数据集上使用平均绝对误差(MAE)和均方误差(MSE)作为评估指标进行了评估。实验结果表明,与现有最先进方法相比,我们的模型实现了优越的性能。此外,我们结合了原则性的贝叶斯推断技术,以提供人群计数预测的不确定性估计,为模型的输出提供置信度度量。

英文摘要:

Crowd counting is a challenging yet critical task in computer vision with applications ranging from public safety to urban planning. Recent advances using Convolutional Neural Networks (CNNs) that estimate density maps have shown significant success. However, accurately counting individuals in highly congested scenes remains an open problem due to severe occlusions, scale variations, and perspective distortions, where people appear at drastically different sizes across the image. In this work, we propose a novel deep learning architecture that effectively addresses these challenges. Our network integrates a ResNet-based feature extractor for capturing rich hierarchical representations, followed by a downsampling block employing dilated convolutions to preserve spatial resolution while expanding the receptive field. An upsampling block using transposed convolutions reconstructs the high-resolution density map. Central to our architecture is a novel Perspective-aware Aggregation Module (PAM) designed to enhance robustness to scale and perspective variations by adaptively aggregating multi-scale contextual information. We detail the training procedure, including the loss functions and optimization strategies used. Our method is evaluated on three widely used benchmark datasets using Mean Absolute Error (MAE) and Mean Squared Error (MSE) as evaluation metrics. Experimental results demonstrate that our model achieves superior performance compared to existing state-of-the-art methods. Additionally, we incorporate principled Bayesian inference techniques to provide uncertainty estimates along with the crowd count predictions, offering a measure of confidence in the model's outputs.

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

↑