首先感谢原作者的源码https://blog.csdn.net/xczexcel/article/details/79405272,才有了这份基于tensorflow的ssd+sort的代码
原来是基于caffe的ssd+sort的,但是在linux下装caffe装的心烦,所以就退而求其次,选择tensorflow版本的ssd,希望一起交流学习。https://github.com/balancap/SSD-Tensorflow (tensorflow版本的ssd目标检测)
如果遇到问题,欢迎留言。
根据上面两个代码,自己修改了对应的检测和关联数据接口,使用视频流作为数据输入,并输出对应的每帧图像到对应目录,同时显示追踪过程到屏幕上。具体可参考对应源码。https://download.csdn.net/download/ycc2011/10861686(我需要一些下载积分,所以设置了最低的一分下载积分。。。)
程序运行截图如下:(不知道是数据集还是数据关联的原因,实际效果不太好) 最后发现是这个tensorflow版本的SSD问题