Skip to content

CGNet:基于组级匹配的弱监督视频个体计数

Xinyan Liu, Guorong Li, Yuankai Qi, Ziheng Yan, Zhenjun Han, Anton van den Hengel, Ming-Hsuan Yang, Qingming Huang. Weakly Supervised Video Individual Counting. CVPR 2024(依据用户提供的论文文件)。主要机构:中国科学院大学、Macquarie University、University of Adelaide、University of California, Merced 等。代码与模型

研究类型:视频理解 / 视频个体计数 / 弱监督学习;涉及多目标关联与无人机视频,但不属于视觉语言导航。下文页码按所给 PDF 页序计。

三个问题

1. 论文解决了什么问题?

视频个体计数(VIC)要统计整段视频出现过多少个不同的人,而不是逐帧人数之和。既有 VIC 与多目标跟踪方法依赖跨帧一致的身份或轨迹标注,标注昂贵,身份切换还会导致重复计数。作者提出弱监督 VIC(WVIC):保留个体位置与相邻采样帧间的进入/离开标注,去掉具体的身份对应关系,研究能否仍准确识别新增个体。

2. 作者解决问题提出了什么方法?

CGNet 把训练目标从“每个个体必须匹配正确身份”放宽为“两帧共有的个体集合能一一匹配”。组级匹配损失 GML 用最优传输求解潜在匹配,以软对比学习拉近共有组、推远进入与离开等不匹配组。推理时,定位器和 ConvNeXt-S 提取目标特征,记忆计数器 MCP 将当前个体与近期多模板记忆匹配,累加首帧人数与后续新增人数。作者还构建了移动无人机计数数据集 UAVVIC。

3. 对我有什么启发?

以下为阅读启发:监督应围绕最终任务所需的最少信息设计。若目标只是总数,不一定需要完整、准确的身份轨迹;“共有组与新增组能否分开”可能已经足够。弱监督信号也不等于只有视频级总数:本文仍需要逐个体位置和进出标记。对移动机器人或无人机的巡检计数,可探索“组级学习 + 有期限记忆”,但需额外验证长时离场后重返是否重复计数。

一句话总结

CGNet 用位置与进出标记代替完整轨迹,通过组级软对比学习和记忆去重完成视频个体计数;在 SenseCrowd、CroHD、UAVVIC 上的 MAE 分别为 8.86、75.0、12.9,均低于对照 DRNet,同时显著减少跨帧关联标注时间(表 2–4、9)。

问题与动机

普通视频人群计数(VCC)输出每帧人数,VIC 则需要跨帧去重。核心计算是“首帧人数 + 后续首次出现的个体数”,而非把每帧人数累加,也不是扣除离开人数。出流虽然不直接进入总数公式,却有助于区分共有与不共有目标。

对一对采样帧,只要识别出哪些个体是新增的,已有个体之间即便没有正确的身份级对应,也不一定影响计数。这一观察使作者将昂贵的逐人匹配改为组级监督:每个人仍有位置标注,但只需说明是否在相邻采样帧中出现过,无需分配跨帧一致 ID(第 1、3 节、图 1)。

方法:组级学习与记忆计数

1. 定位与四组划分

定位器独立利用位置标注训练,默认采用 FIDT,并将骨干由 HRNet-W48 换为 W18。点定位输出以中心裁取 96×64 图块;若输出框,则直接裁框。图块缩放到 224×224,经 ConvNeXt-S 提取、展平并归一化为个体特征。

两帧目标按标注分为前帧共有组 X、前帧离开组、后帧共有组 Y、后帧进入组。其相似度矩阵分成四块:共有组之间为 S0,共有与新增/离开之间为 S1/S2,离开与进入之间为 S3(第 3.1 节、图 2)。

2. 组级匹配损失 GML

对共有组,潜在匹配矩阵 Ω 的行列和均为 1,允许网络自己选择一一对应。软对比项依据 Ω 加权正负对;该优化被写成平衡最优传输问题,以 Sinkhorn 算法求解。训练同时将 S1/S2 的跨组配对作为负样本,帮助区分真正进入或离开的个体。

S3 表示两侧不同个体的相似度,采用带阈值的 hinge 损失,仅惩罚过大的相似度,避免把所有特征强推至零相似而增加退化风险。总损失为软对比项与 hinge 项之和(公式 3–6)。放宽的是训练监督中的身份配对要求,推理仍使用显式匹配。

3. 带生命周期的记忆计数器 MCP

MCP 无需额外训练,为近期出现的目标保留多份外观模板。当前特征与记忆通过 Hungarian 算法做一一匹配,匹配代价超过阈值则判为新增,并创建记忆;成功匹配则补充模板、重置生命周期。目标连续未匹配时生命周期递减,归零后删除。默认采样间隔 3 秒,匹配阈值 0.7,ttl_max=3,记忆模板容量参数 mem_max=5(第 3.2、4.3 节)。

因此记忆能缓解短时遮挡、外观变化和重新进入造成的重复计数,但并非保留全视频身份历史。最终以首帧人数和每次推断的新增数累加输出(公式 10)。

实验与结果

数据与指标

  • SenseCrowd:634 段视频,沿用 DRNet 的训练、验证和测试划分;CroHD:4 段训练视频、5 段测试视频。两者已有轨迹被转换为进出组标注。
  • UAVVIC:移动无人机拍摄的 221 段视频,100/21/100 段用于训练/验证/测试;按 3 秒采样得 5,396 帧、398,158 个框及相邻帧组级标注,涵盖行人、汽车、公交车、厢式车。各类别分别训练评测,并非统一多类计数模型(第 4.1、4.4 节)。
  • 评价:论文报告 MAE、标为 MSE 的误差指标和 WRAE,均越低越好;WRAE 按视频长度加权相对绝对误差,以兼顾不同长度和人数的视频。与多目标跟踪对比时,将整段视频的 ID 数作为预测总数。

主要对比

数据集 方法 MAE ↓ MSE(论文列名)↓ WRAE(%)↓
SenseCrowd DRNet 12.3 24.7 12.7
SenseCrowd CGNet 8.86 17.69 12.6
CroHD DRNet 141.1 192.3 27.4
CroHD CGNet 75.0 95.1 14.5
UAVVIC DRNet 18.4 41.0 15.2
UAVVIC CGNet 12.9 37.4 12.0

SenseCrowd、CroHD 的 MAE 相对 DRNet 下降约 28% 和 47%;UAVVIC 四类汇总 MAE 下降约 30%。但 UAVVIC 没有轨迹标注,因此对照方法在 VisDrone 上训练,CGNet 则利用 UAVVIC 的弱标注,存在训练域和数据来源差异,不能将该提升完全归因于监督形式(表 2–4、第 4.4 节)。

消融与标注成本

  • 去掉 S1/S2 负样本约束,MAE 从完整模型的 8.86 升至 13.35;以普通 L1 替代 hinge 项为 9.84,说明不匹配组和软间隔均有贡献。
  • 去掉 MCP,MAE 为 10.71、WRAE 为 16.9%;完整模型分别为 8.86、12.6%,且视频越长,记忆带来的误差改善越明显,但极高密度下更多模板也可能增加混淆(表 5、图 5)。
  • 对五组不同密度帧对,进出标注用时约为身份关联标注的 30%。例如 116/115 人的帧对为 53.7 秒对 174.4 秒,289/292 人为 95.0 秒对 586.9 秒。这是选定帧对的关联标注成本对比,不等于完整数据采集与位置标注总成本(表 9)。
  • CGNet 还可输出伪轨迹训练现有方法;DRNet 使用其伪轨迹时 MAE 为 14.5,使用真实轨迹时为 12.3,说明较弱标注能生成有用关联,但与真实轨迹仍有差距(表 8)。

局限与批判性阅读

正文没有独立的局限章节。作者在消融中指出:密集场景候选过多会增加匹配混淆;记忆生命周期过长会保存无关模板;采样间隔太大可能遗漏短时出现的人。CroHD15 的固定点裁剪也难覆盖合适目标尺度(第 4.4–4.5 节)。

以下为进一步分析:弱监督仍依赖个体位置、进入/离开判断,严重遮挡下人工标注并不容易;生命周期到期后重新出现的个体可能被当作新增,因而长时间、跨区域唯一身份计数的保证未明确说明。模型上游的漏检和外观特征质量也会直接影响最终总数。

复现还需核对报告口径:表 7 的默认 FIDT 配置 WRAE 为 9.27,与表 2/5 的 12.6 不一致,正文未明确解释;表 6 的 4 秒间隔有最低 MAE,但默认 3 秒有最低 WRAE,因此“最佳间隔”取决于指标。公式 7 对多模板距离使用最大值,代码中的聚合与阈值应另行核验,不能自行改成常见的最小距离匹配。

与已有笔记的联系

与 ATBA 相似,CGNet 先重新审视任务需要哪些监督,再设计中间匹配;前者以少量动作边界替代逐帧对齐,后者以共有/进出组替代真实身份对应。与 HUMEMBR 的跨天身份记忆不同,MCP 是服务于计数的近期外观缓存,不负责人物语义、作息或语言查询。

术语与关键证据

  • VIC / WVIC:视频不同个体总数统计及其无身份轨迹标注版本,区别于逐帧人群计数。
  • 组级匹配:知道两帧共有集合,但不监督集合内部每对真实身份;见图 1–2、公式 3–6。
  • MCP:依靠近期多模板与生命周期更新进行新增判断的计数器;见第 3.2 节。
  • 关键证据:表 2–4 比较总数误差;表 5、图 5 分析组件与记忆;表 9 对照人工关联标注时间。

返回视频计数目录 · 返回 Paper Reading