From 562cec6814aba6a8db72178c3e6652089b210930 Mon Sep 17 00:00:00 2001 From: ZhongdaoWang Date: Tue, 1 Oct 2019 09:32:20 +0800 Subject: [PATCH 1/2] Update README.md --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 6c7fe69..766eba0 100644 --- a/README.md +++ b/README.md @@ -1,9 +1,10 @@ # Towards-Realtime-MOT +Still in progress, will update constantly, thank you for your attention! ## Introduction This repo is the a codebase of the Joint Detection and Embedding (JDE) model. JDE is a fast and high-performance multiple-object tracker that learns the object detection task and appearance embedding task simutaneously in a shared neural network. Techical details are described in our [arXiv preprint paper](https://arxiv.org). By using this repo, you can simply achieve **MOTA 64%+** on the "private" protocol of [MOT-16 challenge](https://motchallenge.net/tracker/JDE), and with a near real-time speed at **18~24 FPS** (Note this speed is for the entire system, including the detection step! ) . -We hope this repo will help researches/engineers to develop more practical MOT systems. For algorithm development, we provide training data, baseline models and evaluation methods to make a level playground. For application usage, we also provide a small video demo that takes a raw video as input without any bells and whistles. +We hope this repo will help researches/engineers to develop more practical MOT systems. For algorithm development, we provide training data, baseline models and evaluation methods to make a level playground. For application usage, we also provide a small video demo that takes raw videos as input without any bells and whistles. ## Installation From b90e8b8b4bea9d61c7e4f0f4bbc9c13fa00d5869 Mon Sep 17 00:00:00 2001 From: ZhongdaoWang Date: Tue, 1 Oct 2019 21:46:21 +0800 Subject: [PATCH 2/2] Update README.md --- README.md | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 766eba0..dace4aa 100644 --- a/README.md +++ b/README.md @@ -6,7 +6,13 @@ This repo is the a codebase of the Joint Detection and Embedding (JDE) model. JD We hope this repo will help researches/engineers to develop more practical MOT systems. For algorithm development, we provide training data, baseline models and evaluation methods to make a level playground. For application usage, we also provide a small video demo that takes raw videos as input without any bells and whistles. -## Installation +## Requirements +* Python 3.6 +* [Pytorch](https://pytorch.org) >= 1.0.1 +* [syncbn](https://github.com/ytoon/Synchronized-BatchNorm-PyTorch) (Optional, compile and place it under utils/syncbn, or simply replace with nn.BatchNorm [here](https://github.com/Zhongdao/Towards-Realtime-MOT/blob/master/models.py#L12)) +* [maskrcnn-benchmark](https://github.com/facebookresearch/maskrcnn-benchmark) (Their GPU NMS is used in this project) +* python-opencv +* ffmpeg (Optional, used in the video demo) ## Video Demo @@ -20,3 +26,5 @@ We hope this repo will help researches/engineers to develop more practical MOT s ## Train with custom datasets +## Acknowledgement +A large portion of code is borrowed from [ultralytics/yolov3](https://github.com/ultralytics/yolov3) and [longcw/MOTDT](https://github.com/longcw/MOTDT), many thanks to their wonderful work!