ConsistencyTrack: A Robust Multi-Object Tracker with a Generation Strategy of Consistency Model

Lifan Jiang, Zhihui Wang, Siqi Yin, Guangxiao Ma, Peng Zhang +1 more
8/28/2024
cs.CV

Official code: Tankowa/ConsistencyTrack (10 stars, last commit 2024-08-23).

Abstract

Multi-object tracking (MOT) is a critical technology in computer vision, designed to detect multiple targets in video sequences and assign each target a unique ID per frame. Existed MOT methods excel at accurately tracking multiple objects in real-time across various scenarios. However, these methods still face challenges such as poor noise resistance and frequent ID switches. In this research, we propose a novel ConsistencyTrack, joint detection and tracking(JDT) framework that formulates detection and association as a denoising diffusion process on perturbed bounding boxes. This progressive denoising strategy significantly improves the model's noise resistance. During the training phase, paired object boxes within two adjacent frames are diffused from ground-truth boxes to a random distribution, and then the model learns to detect and track by reversing this process. In inference, the model refines randomly generated boxes into detection and tracking results through minimal denoising steps. ConsistencyTrack also introduces an innovative target association strategy to address target occlusion. Experiments on the MOT17 and DanceTrack datasets demonstrate that ConsistencyTrack outperforms other compared methods, especially better than DiffusionTrack in inference speed and other performance metrics. Our code is available at https://github.com/Tankowa/ConsistencyTrack.

View on arXivView PDF

Code Implementations(1)

101C++, PythonAug 19, 20242 years agoNOASSERTION

Similar Papers

Cite this paper

@article{jiang2024consistencytrack,
  title  = {ConsistencyTrack: A Robust Multi-Object Tracker with a Generation Strategy of Consistency Model},
  author = {Lifan Jiang and Zhihui Wang and Siqi Yin and Guangxiao Ma and Peng Zhang and Boxi Wu},
  year   = {2024},
  eprint = {2408.15548},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2408.15548v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2408.15548.svg)](https://codeofpaper.com/paper/2408.15548)

Discussion