3 个月前

SparseTrack:基于伪深度的场景分解实现多目标追踪

SparseTrack:基于伪深度的场景分解实现多目标追踪

摘要

探索鲁棒且高效的关联方法一直是多目标跟踪(Multiple-Object Tracking, MOT)领域的核心问题。尽管现有跟踪方法已取得令人瞩目的性能,但在密集场景中,目标之间的拥堵与频繁遮挡仍是制约性能提升的关键挑战。本文揭示,对密集场景进行稀疏分解是提升被遮挡目标关联性能的关键步骤。为此,我们提出一种伪深度估计方法,仅从二维图像中推断目标之间的相对深度信息。其次,我们设计了一种深度级联匹配(Depth Cascading Matching, DCM)算法,利用获取的深度信息将密集的目标集合分解为多个稀疏的目标子集,并按照从近到远的顺序依次对这些稀疏子集进行数据关联。通过将伪深度估计方法与DCM匹配策略融合至数据关联流程中,我们提出了一种新型跟踪器——SparseTrack。SparseTrack为解决复杂密集场景下的多目标跟踪问题提供了全新的视角。仅采用IoU匹配策略,SparseTrack在MOT17和MOT20基准测试上即达到了与当前最先进(State-of-the-Art, SOTA)方法相当的性能。代码与模型已公开发布于:\url{https://github.com/hustvl/SparseTrack}。

代码仓库

hustvl/sparsetrack
官方
pytorch
GitHub 中提及
Robotmurlock/Motrack
GitHub 中提及

基准测试

基准方法指标
multi-object-tracking-on-dancetrackSparseTrack
AssA: 39.3
DetA: 79.2
HOTA: 55.7
IDF1: 58.1
MOTA: 91.3
multi-object-tracking-on-mot17SparseTrack
HOTA: 65.1
IDF1: 80.1
MOTA: 81.0
multi-object-tracking-on-mot20-1SparseTrack
HOTA: 63.4
IDF1: 77.3
MOTA: 78.2

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SparseTrack:基于伪深度的场景分解实现多目标追踪 | 论文 | HyperAI超神经