YOLO11-JDE: Fast and Accurate Multi-Object Tracking with Self-Supervised Re-ID
arXiv:2501.13710 · doi:10.1109/WACVW65960.2025.00092
Abstract
We introduce YOLO11-JDE, a fast and accurate multi-object tracking (MOT) solution that combines real-time object detection with self-supervised Re-Identification (Re-ID). By incorporating a dedicated Re-ID branch into YOLO11s, our model performs Joint Detection and Embedding (JDE), generating appearance features for each detection. The Re-ID branch is trained in a fully self-supervised setting while simultaneously training for detection, eliminating the need for costly identity-labeled datasets. The triplet loss, with hard positive and semi-hard negative mining strategies, is used for learning discriminative embeddings. Data association is enhanced with a custom tracking implementation that successfully integrates motion, appearance, and location cues. YOLO11-JDE achieves competitive results on MOT17 and MOT20 benchmarks, surpassing existing JDE methods in terms of FPS and using up to ten times fewer parameters. Thus, making our method a highly attractive solution for real-world applications.
This paper has been accepted to the 5th Workshop on Real-World Surveillance: Applications and Challenges (WACV 2025)
References in corpus (21)
- FaceNet: A Unified Embedding for Face Recognition and Clustering
- YOLOv4: Optimal Speed and Accuracy of Object Detection
- A Simple Framework for Contrastive Learning of Visual Representations
- Representation Learning with Contrastive Predictive Coding
- mixup: Beyond Empirical Risk Minimization
- Simple Online and Realtime Tracking
- YOLOX: Exceeding YOLO Series in 2021
- In Defense of the Triplet Loss for Person Re-Identification
- FairMOT: On the Fairness of Detection and Re-Identification in Multiple Object Tracking
- MOT16: A Benchmark for Multi-Object Tracking
- HOTA: A Higher Order Metric for Evaluating Multi-Object Tracking
- Deep Learning in Video Multi-Object Tracking: A Survey
- MOT20: A benchmark for multi object tracking in crowded scenes
- CrowdHuman: A Benchmark for Detecting Human in a Crowd
- BoT-SORT: Robust Associations Multi-Pedestrian Tracking
- SoccerNet-Tracking: Multiple Object Tracking Dataset and Benchmark in Soccer Videos
- FastReID: A Pytorch Toolbox for General Instance Re-identification
- DEFT: Detection Embeddings for Tracking
- In Pursuit of Many: A Review of Modern Multiple Object Tracking Systems
- Online and Real-Time Tracking in a Surveillance Scenario
- Temporal Correlation Meets Embedding: Towards a 2nd Generation of JDE-based Real-Time Multi-Object Tracking