今天介绍一份PyTorch实现的多目标跟踪的开源库,在Github 上搜索 MOT tracking ,该份代码排名第一位,目前已经有505颗星。
该代码实现的是广为人知的deep sort多目标跟踪算法,算法原作者的实现是基于TensorFlow的,作者用PyTorch实现了RE-ID(目标重识别)模块,并将目标检测模型从Faster RCNN 替换为YOLOv3。
作者提供了预训练模型,不过如果你要自己训练RE-ID模型的话也是支持的。
运行结果示例:
作者在视频中跑的结果:
效果看起来还是蛮好的,大家也可以玩试试~
代码地址:
https://github.com/ZQPei/deep_sort_pytorch