AI学习者 · 2020年11月24日

简单粗暴的多对象目标跟踪神器 – DeepSort

文章转载于微信公众号:OpenCV学堂
作者:gloomyfish

DeepSort

对象跟踪问题一直是计算机视觉的热点任务之一,简单的可以分为单目标跟踪与多目标跟踪,最常见的目标跟踪算法都是基于检测的跟踪算法,首先发现然后标记,好的跟踪算法必须具备REID的能力。今天小编斗胆给大家推荐一个结合传统算法跟深度学习,特别好用的对象跟踪算法框架DeepSort

DeepSort的核心思想主要分为两块,一块可以简单称为Deep,另外一个可以称为Sort,背后的算法支持分别基于深度学习模型与卡尔曼滤波,是典型的结合深度学习与传统方法的混合算法框架实现了比较稳定的跟踪效果。见得工作原理示意图如下:

image.png
从输入视频流开始,首先通过对象检测算(YOLOv3)法实现对象检测,然后基于检测结果标记利用DeepSort实现跟踪。

Deepsort的相关论文如下:

https://arxiv.org/abs/1703.07402

Pytorch版本的代码实现如下:

https://github.com/ZQPei/deep_sort_pytorch

代码演示

获取代码

git clone https://github.com/ZQPei/deep_sort_pytorch

获取代码之后,还需要下载YOLOv3模型与Deep的t7模型,分别是

- yolov3.weights

然后 运行下面命令行

python yolov3_deepsort.py D:\images\video\TownCentreXVID.avi –display

运行结果如下:

image.png
我只能说效果绝对靠谱!在我的1050Ti笔记本上测试通过!
君子藏器于身,待时而动

推荐阅读

推荐阅读
关注数
18790
内容数
1342
嵌入式端AI,包括AI算法在推理框架Tengine,MNN,NCNN,PaddlePaddle及相关芯片上的实现。欢迎加入微信交流群,微信号:aijishu20(备注:嵌入式)
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息