PyTorch 多目标跟踪

今天介绍一份PyTorch实现的多目标跟踪的开源库,在Github 上搜索 MOT tracking ,该份代码排名第一位,目前已经有505颗星

该代码实现的是广为人知的deep sort多目标跟踪算法,算法原作者的实现是基于TensorFlow的,作者用PyTorch实现了RE-ID(目标重识别)模块,并将目标检测模型从Faster RCNN 替换为YOLOv3

作者提供了预训练模型,不过如果你要自己训练RE-ID模型的话也是支持的。

PyTorch 多目标跟踪

运行结果示例:

PyTorch 多目标跟踪

作者在视频中跑的结果:

效果看起来还是蛮好的,大家也可以玩试试~

代码地址:

https://github.com/ZQPei/deep_sort_pytorch

 PyTorch 多目标跟踪

推荐阅读

人工智能与算法学习--近期合集

PyTorch 多目标跟踪

PyTorch 多目标跟踪

“哪吒头”—玩转小潮流