
摘要
我们提出了一种基于求解最优传输问题的新方法,用于处理长视频中的动作分割任务。通过将时间一致性先验编码到一个Gromov-Wasserstein问题中,我们能够从视频帧与动作类别之间的噪声亲和力/匹配成本矩阵中解码出时间上一致的分割结果。与以往的方法不同,我们的方法不需要预先知道视频中的动作顺序即可实现时间一致性。此外,我们所得到的(融合)Gromov-Wasserstein问题可以通过在GPU上进行几次投影镜像下降迭代来高效求解。我们在无监督学习环境中展示了该方法的有效性,其中该方法被用于生成伪标签以进行自训练。我们在Breakfast、50-Salads、YouTube Instructions和Desktop Assembly数据集上评估了我们的分割方法和无监督学习流程,取得了当前最先进的无监督视频动作分割任务结果。
代码仓库
mingu6/action_seg_ot
官方
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| unsupervised-action-segmentation-on-breakfast | ASOT | Acc: 56.1 F1: 38.3 JSD: 94.9 Precision: 36.7 Recall: 40.1 mIoU: 18.6 |
| unsupervised-action-segmentation-on-ikea-asm | ASOT | Accuracy: 34.0 F1: 27.9 JSD: 88.7 Precision: 21.1 Recall: 24.0 |
| unsupervised-action-segmentation-on-youtube | ASOT | Acc: 52.9 F1: 35.1 Precision: 47.6 Recall: 27.8 mIoU: 24.7 |