PyTorch 多目标跟踪


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

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

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

运行结果示例:

作者在视频中跑的结果:

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

代码地址:

https://github.com/ZQPei/deep_sort_pytorch


目标跟踪交流群

(0)

相关推荐