diff --git a/README.md b/README.md index 4ac295c..cff3936 100644 --- a/README.md +++ b/README.md @@ -10,12 +10,14 @@ We hope this repo will help researches/engineers to develop more practical MOT s ## 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) +* [Pytorch](https://pytorch.org) >= 1.2.0 * python-opencv -* ffmpeg (Optional, used in the video demo) -* [py-motmetrics](https://github.com/cheind/py-motmetrics) (Simply `pip install motmetrics`) +* [py-motmetrics](https://github.com/cheind/py-motmetrics) (`pip install motmetrics`) +* cython-bbox (`pip install cython_bbox`) +* (Optional) ffmpeg (used in the video demo) +* (Optional) [syncbn](https://github.com/ytoon/Synchronized-BatchNorm-PyTorch) (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)~~ + ## Video Demo