【ZMK】CFnet目标跟踪算法Matlab代码运行指南

由于最近在做相关滤波方面的工作。

CFnet是文章End-to-end representation learning for Correlation Filter based tracking中算法的简称。

该算法页面在https://github.com/bertinetto/cfnet

本人所用笔记本电脑配置为Win10,显卡GTX 965M。

软件为Matlab R2016a,matconvnet-1.0-beta24,CUDA8.0.44_win10+cudnn-8.0-windows10-x64-v5.0

matconvnet安装配置我是根据Kindle君来安装的,具体地址请参考:

https://blog.****.net/yexiaogu1104/article/details/73177690

前期准备好了以后,从算法页面把算法下载下来,解压后我放置在G盘

【ZMK】CFnet目标跟踪算法Matlab代码运行指南

然后根据第三步骤,下载先前训练好的网络,下载后放置在pretrained文件下。

我是在pretrained文件下又创立一个networks文件夹,把预先训练好的网络放进去。

【ZMK】CFnet目标跟踪算法Matlab代码运行指南

然后根据第四步骤,在cfnet/src/tracking/文件下把env_paths_tracking.m.examplestartup.m.example中的example去掉,变为m文件。打开env_paths_tracking.m文件,修改里面的路径。

【ZMK】CFnet目标跟踪算法Matlab代码运行指南

第一个就是预训练网络的路径,第二个就是要测试的数据的路径

第三个ILSVRC2015.stats.mat在我是直接用了该作者前期siamese-fc算法文件中的,大家可以自行下载,路径添加正确就行了。

【ZMK】CFnet目标跟踪算法Matlab代码运行指南

然后是startup.m文件路径的修改,就是把matconvnet路径放进去就行了。

第五步就是数据集准备,按照作者给的地址下载数据集,然后解压后放到data下面。

data文件夹下面记得要创立个validation文件夹,然后把各个子文件放进去。

【ZMK】CFnet目标跟踪算法Matlab代码运行指南

然后打开run_*_evaluation.m,我是run_cfnet1_evaluation.m

修改最后一行,把“all”改成你要跑的数据集的名字。

另外,跑出来没有图像显示,需要自行添加一行tracker_par.visualization = true;

【ZMK】CFnet目标跟踪算法Matlab代码运行指南

然后运行程序,结果如下所示

【ZMK】CFnet目标跟踪算法Matlab代码运行指南

然后我用自己的数据集跑了一下,效果还不错。用的鱼眼摄像头,自己矫正。

【ZMK】CFnet目标跟踪算法Matlab代码运行指南

【ZMK】CFnet目标跟踪算法Matlab代码运行指南

记得一点,自己的数据集,要把groundtruth自己标定好。我是五帧标定一次。第一次写就写这么多,有疑问的欢迎大家留言。