arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2311.08043cs.CV

基于对比学习的 Transformer 多目标跟踪

Contrastive Learning for Multi-Object Tracking with Transformers

KU Leuven(鲁汶大学);TRACE vzw

Pierre-François De Plaen, Nicola Marinello, Marc Proesmans, Tinne Tuytelaars, Luc Van Gool

AI总结:

本文提出用实例级对比损失、改进采样策略和轻量级分配方法将 DETR 改造为多目标跟踪模型,以低开销在 BDD100K 上提升 2.6 mMOTA,并在 MOT17 上达到可比性能。

摘要:

DEtection TRansformer(DETR)通过将目标检测建模为转换任务(即把图像特征转换为对象级表示),为目标检测开辟了新的可能性。以往工作通常会向 DETR 添加昂贵的模块来执行多目标跟踪(MOT),导致架构更加复杂。与之不同,我们展示了如何通过采用实例级对比损失、改进的采样策略和轻量级分配方法,将 DETR 转化为 MOT 模型。我们的训练方案能够在学习目标外观的同时保持检测能力,并且开销很小。其性能在具有挑战性的 BDD100K 数据集上比此前的最优水平高出 +2.6 mMOTA,并在 MOT17 数据集上与现有基于 Transformer 的方法相当。

英文摘要:

The DEtection TRansformer (DETR) opened new possibilities for object detection by modeling it as a translation task: converting image features into object-level representations. Previous works typically add expensive modules to DETR to perform Multi-Object Tracking (MOT), resulting in more complicated architectures. We instead show how DETR can be turned into a MOT model by employing an instance-level contrastive loss, a revised sampling strategy and a lightweight assignment method. Our training scheme learns object appearances while preserving detection capabilities and with little overhead. Its performance surpasses the previous state-of-the-art by +2.6 mMOTA on the challenging BDD100K dataset and is comparable to existing transformer-based methods on the MOT17 dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑