论文标题

称重计数:通过加强学习的顺序人群计数

Weighing Counts: Sequential Crowd Counting by Reinforcement Learning

论文作者

Liu, Liang, Lu, Hao, Zou, Hongwei, Xiong, Haipeng, Cao, Zhiguo, Shen, Chunhua

论文摘要

我们将算作依次的决策问题提出,并提出一个新颖的人群来计算可通过深入的增强学习来解决的模型。与直接输出计数值的现有计数模型相反,我们将一步估计分为一个更容易且更可拖延的子否决问题的顺序。这种顺序决策性质完全对应于现实尺度称重的物理过程。受尺度称重的启发,我们提出了一种新颖的“计数规模”,称为库libranet,其中计数值用重量类似。通过将人群图像几乎放在秤的一侧,libranet(Agent)依次学习将适当的权重放在另一侧以匹配人群数。在每个步骤中,Libranet根据当前的人群图像特征和放置在比例锅上的重量(状态)从重量框(预定义的动作池)中选择一个重量(动作)。必须根据针的反馈(Q值)来学会平衡尺度。我们表明,图书馆通过可视化决策过程如何选择动作来确切地实现量表称量。广泛的实验证明了我们的设计选择的有效性,并报告了一些人群计算基准的最新结果。我们还展示了库层的良好的跨数据集概括。代码和型号可在以下网址提供:https://git.io/libranet

We formulate counting as a sequential decision problem and present a novel crowd counting model solvable by deep reinforcement learning. In contrast to existing counting models that directly output count values, we divide one-step estimation into a sequence of much easier and more tractable sub-decision problems. Such sequential decision nature corresponds exactly to a physical process in reality scale weighing. Inspired by scale weighing, we propose a novel 'counting scale' termed LibraNet where the count value is analogized by weight. By virtually placing a crowd image on one side of a scale, LibraNet (agent) sequentially learns to place appropriate weights on the other side to match the crowd count. At each step, LibraNet chooses one weight (action) from the weight box (the pre-defined action pool) according to the current crowd image features and weights placed on the scale pan (state). LibraNet is required to learn to balance the scale according to the feedback of the needle (Q values). We show that LibraNet exactly implements scale weighing by visualizing the decision process how LibraNet chooses actions. Extensive experiments demonstrate the effectiveness of our design choices and report state-of-the-art results on a few crowd counting benchmarks. We also demonstrate good cross-dataset generalization of LibraNet. Code and models are made available at: https://git.io/libranet

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源