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)