3 个月前

FairMOT:多目标跟踪中检测与重识别的公平性研究

FairMOT:多目标跟踪中检测与重识别的公平性研究

摘要

多目标跟踪(Multi-object tracking, MOT)是计算机视觉领域的一个重要问题,具有广泛的应用前景。将MOT建模为单一网络中目标检测与重识别(re-ID)的多任务学习,具有显著吸引力,因为它能够实现两个任务的联合优化,并具备较高的计算效率。然而,我们发现这两个任务之间存在明显的竞争关系,需加以妥善处理。特别是,以往方法通常将re-ID视为次要任务,其性能严重依赖于主要的检测任务,导致网络倾向于优化检测任务,从而对re-ID任务产生不公平的偏差。为解决该问题,本文提出一种简单而有效的框架——FairMOT,其基于无锚点(anchor-free)的目标检测架构CenterNet。需要强调的是,FairMOT并非简单地将CenterNet与re-ID进行拼接,而是通过一系列经过充分实证研究验证的关键设计,显著提升了跟踪性能。所提出的方法在检测与跟踪任务上均实现了高精度,且在多个公开数据集上显著超越了当前最先进方法。相关源代码与预训练模型已开源,地址为:https://github.com/ifzhang/FairMOT。

代码仓库

ydhcg-bobo/stcmot
pytorch
GitHub 中提及
ankitsinghsuraj/mot20
pytorch
GitHub 中提及
IMBINGO95/FairMOT
pytorch
GitHub 中提及
nolanzzz/MTMCT
pytorch
GitHub 中提及
harsh2912/people-tracking
pytorch
GitHub 中提及
15534081591/FairMOT
mindspore
GitHub 中提及
nemonameless/fairmot
pytorch
GitHub 中提及
dhu-mmct/dhu-mmct
pytorch
GitHub 中提及
oljikeboost/PlayerTracking
pytorch
GitHub 中提及
zengwbz/Face-Tracking-usingFairMOT
pytorch
GitHub 中提及
gsan2/FairMOT
pytorch
GitHub 中提及
ifzhang/FairMOT
官方
pytorch
GitHub 中提及
HoganZhang/FairMOT
pytorch
GitHub 中提及
FlorentijnD/FairMOT
pytorch
GitHub 中提及
microsoft/FairMOT
pytorch
GitHub 中提及
microsoft/UDA
pytorch
GitHub 中提及
Bangbangbanana/fairmot_mindspore
mindspore
GitHub 中提及
cds-mipt/cds-tracking
pytorch
GitHub 中提及
lilin19890401/FairMOT
pytorch
GitHub 中提及
dingwoai/FairMOT-BDD100K
pytorch
GitHub 中提及

基准测试

基准方法指标
multi-object-tracking-on-2dmot15-1FairMOT
MOTA: 60.6
multi-object-tracking-on-dancetrackFairMOT
AssA: 23.8
DetA: 66.7
HOTA: 39.7
IDF1: 40.8
MOTA: 82.2
multi-object-tracking-on-hieveFairMOT
MOTA: 35.0
multi-object-tracking-on-mot16FairMOT
MOTA: 74.9
multi-object-tracking-on-mot17FairMOT
IDF1: 72.3
MOTA: 73.7
multi-object-tracking-on-mot20-1FairMOT
IDF1: 67.3
MOTA: 61.8
multi-object-tracking-on-sportsmotFairMOT
AssA: 34.7
DetA: 70.2
HOTA: 49.3
IDF1: 53.5
MOTA: 86.4
multiple-object-tracking-on-sportsmotFairMOT
AssA: 34.7
DetA: 70.2
HOTA: 49.3
IDF1: 53.5
MOTA: 86.4

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
FairMOT:多目标跟踪中检测与重识别的公平性研究 | 论文 | HyperAI超神经