LLM2D
人群计数掩码自动补全:用于鲁棒人群密度预测
CrowdMAC: Masked Crowd Density Completion for Robust Crowd Density Forecasting
作者: Ryo Fujii, Ryo Hachiuma, Hideo Saito
发布日期: 11/27/2024
arXiv ID: oai:arXiv.org:2407.14725v2

摘要

人群密度预测任务旨在根据观察到的过去人群密度图预测未来人群密度图的变化。然而,由于行人漏检,过去的 人群密度图往往不完整,因此开发一个能够抵抗漏检的鲁棒人群密度预测模型至关重要。本文提出了一种用于人群密度预测的基于掩码的人群密度补全框架 (CrowdMAC),该框架同时训练从部分掩码的过去人群密度图(即从存在漏检的过去地图预测地图)预测未来人群密度图,同时重建掩码的观测图(即填充存在漏检的过去地图)。此外,我们提出了时间密度感知掩码 (TDM),它考虑人群密度图的稀疏性和后续帧对预测任务的信息量,对观察到的人群密度图中的标记进行非均匀掩码。此外,我们引入了多任务掩码以提高训练效率。在实验中,CrowdMAC 在七个大型数据集(包括 SDD、ETH-UCY、inD、JRDB、VSCrowd、FDST 和 croHD)上取得了最先进的性能。我们还证明了该方法对合成漏检和真实漏检的鲁棒性。