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