HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Learning a Neural Solver for Multiple Object Tracking

{ Laura Leal-Taixe Guillem Braso}

Learning a Neural Solver for Multiple Object Tracking

Abstract

Graphs offer a natural way to formulate Multiple Object Tracking (MOT) within the tracking-by-detection paradigm. However, they also introduce a major challenge for learning methods, as defining a model that can operate on such structured domain is not trivial. As a consequence, most learning-based work has been devoted to learning better features for MOT and then using these with well-established optimization frameworks. In this work, we exploit the classical network flow formulation of MOT to define a fully differentiable framework based on Message Passing Networks (MPNs). By operating directly on the graph domain, our method can reason globally over an entire set of detections and predict final solutions. Hence, we show that learning in MOT does not need to be restricted to feature extraction, but it can also be applied to the data association step. We show a significant improvement in both MOTA and IDF1 on three publicly available benchmarks. Our code is available at https://bit.ly/motsolv.

Benchmarks

BenchmarkMethodologyMetrics
multi-object-tracking-on-2d-mot-2015MPNTrack
IDF1: 58.6
MOTA: 51.5
multi-object-tracking-on-mot16MPNTrack
IDF1: 61.7
MOTA: 58.6
multi-object-tracking-on-mot17MPNTrack
IDF1: 61.7
MOTA: 58.8
multi-object-tracking-on-mot20-1MPNTrack
IDF1: 59.1
MOTA: 57.6

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Learning a Neural Solver for Multiple Object Tracking | Papers | HyperAI