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

学习用于人群计数的判别性特征

Learning Discriminative Features for Crowd Counting

School of Automation Science and Engineering, Xi’an Jiaotong University(西安交通大学自动化科学与工程学院);Institute of Artificial Intelligence and Robotics, Xi’an Jiaotong University(西安交通大学人工智能与机器人研究所);Baidu Research(百度研究院)

Yuehai Chen, Qingzhong Wang, Jing Yang, Badong Chen, Haoyi Xiong, Shaoyi Du

AI总结:

针对拥挤场景人群计数中定位弱和前景背景难分的问题,提出含MPM与CLM的即插即用判别特征学习框架,以增强小目标定位和前景区分能力。

摘要:

高度拥挤区域中的人群计数模型面临两大挑战:定位能力较弱,以及难以区分前景与背景,从而导致估计不准确。其原因在于,高度拥挤区域中的目标通常很小,而卷积神经网络提取的高层特征在表示小目标时判别性不足。为解决这些问题,我们提出了一个用于人群计数的判别性特征学习框架,该框架由掩码特征预测模块(MPM)和有监督像素级对比学习模块(CLM)组成。MPM随机掩码特征图中的特征向量,然后对其进行重建,使模型能够学习被掩码区域中存在的内容,并提升模型在高密度区域定位目标的能力。CLM在特征空间中将目标彼此拉近,并将它们推离背景,使模型能够区分前景目标与背景。此外,所提出的模块还可有益于多种计算机视觉任务,例如人群计数和目标检测;在这些任务中,密集场景或杂乱环境会给准确定位带来挑战。所提出的两个模块均为即插即用,将其整合到现有模型中,有可能提升这些模型在此类场景下的性能。

英文摘要:

Crowd counting models in highly congested areas confront two main challenges: weak localization ability and difficulty in differentiating between foreground and background, leading to inaccurate estimations. The reason is that objects in highly congested areas are normally small and high level features extracted by convolutional neural networks are less discriminative to represent small objects. To address these problems, we propose a learning discriminative features framework for crowd counting, which is composed of a masked feature prediction module (MPM) and a supervised pixel-level contrastive learning module (CLM). The MPM randomly masks feature vectors in the feature map and then reconstructs them, allowing the model to learn about what is present in the masked regions and improving the model's ability to localize objects in high density regions. The CLM pulls targets close to each other and pushes them far away from background in the feature space, enabling the model to discriminate foreground objects from background. Additionally, the proposed modules can be beneficial in various computer vision tasks, such as crowd counting and object detection, where dense scenes or cluttered environments pose challenges to accurate localization. The proposed two modules are plug-and-play, incorporating the proposed modules into existing models can potentially boost their performance in these scenarios.

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

↑