中文

TransTrack:基于 Transformer 的多目标跟踪

计算机视觉与模式识别 2021-05-05 v2

摘要

在这项工作中,我们提出 TransTrack,一种简单而高效的方案来解决多目标跟踪问题。TransTrack 利用了 transformer 架构,即一种基于注意力的查询-键机制。它将前一帧中的目标特征作为当前帧的查询,并引入一组学习得到的目标查询以实现对新出现目标的检测。它通过一次性完成目标检测与目标关联,构建了一种新颖的联合检测与跟踪范式,简化了基于检测跟踪方法中复杂的多步设置。在 MOT17 和 MOT20 基准上,TransTrack 分别取得了 74.5% 和 64.5% 的 MOTA,与最先进的方法具有竞争力。我们期望 TransTrack 为多目标跟踪提供新颖的视角。代码可在:\url{https://github.com/PeizeSun/TransTrack} 获取。

关键词

引用

@article{arxiv.2012.15460,
  title  = {TransTrack: Multiple Object Tracking with Transformer},
  author = {Peize Sun and Jinkun Cao and Yi Jiang and Rufeng Zhang and Enze Xie and Zehuan Yuan and Changhu Wang and Ping Luo},
  journal= {arXiv preprint arXiv:2012.15460},
  year   = {2021}
}

备注

update MOT17 and MOT20