点击我爱计算机视觉标星,更快获取CVML新技术
今天介绍一份PyTorch实现的多目标跟踪的开源库,在Github 上搜索 MOT tracking ,该份代码排名第一位,目前已经有505颗星。
该代码实现的是广为人知的deep sort多目标跟踪算法,算法原作者的实现是基于TensorFlow的,作者用PyTorch实现了RE-ID(目标重识别)模块,并将目标检测模型从Faster RCNN 替换为YOLOv3。
作者提供了预训练模型,不过如果你要自己训练RE-ID模型的话也是支持的。
运行结果示例:
作者在视频中跑的结果:
效果看起来还是蛮好的,大家也可以玩试试~
代码地址:
https://github.com/ZQPei/deep_sort_pytorch
目标跟踪交流群
关注最新最前沿的单目标、多目标跟踪技术,扫码添加CV君拉你入群,(如已为CV君其他账号好友请直接私信)
(请务必注明:跟踪)
喜欢在QQ交流的童鞋,可以加52CV官方QQ群:805388940。
(不会时时在线,如果没能及时通过验证还请见谅)
长按关注我爱计算机视觉