4 个月前

TransTrack: 基于Transformer的多目标跟踪

TransTrack: 基于Transformer的多目标跟踪

摘要

在本研究中,我们提出了一种简单但高效的方案——TransTrack,用于解决多目标跟踪问题。TransTrack 利用了基于注意力机制的变压器架构(transformer architecture),将前一帧中的目标特征作为当前帧的查询(query),并引入一组学习到的目标查询,以实现对新出现目标的检测。它通过一次性完成目标检测和目标关联,建立了一种新颖的联合检测与跟踪范式,简化了基于检测的跟踪方法中复杂的多步骤设置。在 MOT17 和 MOT20 基准测试中,TransTrack 分别达到了 74.5% 和 64.5% 的 MOTA 指标,与现有最先进方法具有竞争力。我们期望 TransTrack 能为多目标跟踪提供一种新的视角。代码已开源,可访问以下链接获取:\url{https://github.com/PeizeSun/TransTrack}。

代码仓库

PeizeSun/TransTrack
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
multi-object-tracking-on-dancetrackTransTrack
AssA: 27.5
DetA: 72.1
HOTA: 45.7
IDF1: 44.8
MOTA: 83.0
multi-object-tracking-on-sportsmotTransTrack
AssA: 57.5
DetA: 82.7
HOTA: 68.9
IDF1: 71.5
MOTA: 92.6

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
TransTrack: 基于Transformer的多目标跟踪 | 论文 | HyperAI超神经