MDNet代码链接:py-MDNet
云GPU:featurize
训练部分
- 对于源代码有两个结果一个是在VOT2013-2015数据集上训练的结果一个是在imagenet-VID上训练的结果
- 先在VOT2013-2015进行训练,数据集下载地址可以参考这个博客(27条消息) 常见目标跟踪数据集下载链接整理(更新中)_laizi_laizi的博客-CSDN博客_目标跟踪数据集下载
- 根据readme保存下载的数据集,因为我在云服务器上跑的(云服务器选择的是featurize这个平台,相对比较平价好用,显卡选用的是RTX 3060)所以数据集的存放地址和readme里面的不一样,在运行代码时需要对一些路径进行修改,改成自己的路径就可以
- 修改prepro_vot.py的路径,其他文件的路径也需要进行相应的修改
#seq_home = 'datasets/VOT'
seq_home='/home/featurize/data'
seqlist_path = '/home/featurize/work/py-MDNet-master/datasets/list/vot-otb.txt'
output_path = '/home/featurize/work/py-MDNet-master/pretrain/data/vot-otb.pkl'
- 然后运行prepro_vot.py文件
运行中出现的问题
出现这个问题的主要原因是vot2015的数据集的图片存放在color目录下,所以需要对代码进行修改
gt = np.loadtxt(os.path.join(seq_home, seq, 'groundtruth.txt'