把STMTrack跑起来（STMTrack: Template-free Visual Tracking with Space-time Memory Networks-CVPR2021）

STMTrack: Template-free Visual Tracking with Space-time Memory Networks

论文连接：https://arxiv.org/pdf/2104.00324v2.pdf
项目连接：https://github.com/fzh0917/STMTrack

环境配置

1. 下载项目文件
   git clone
2. 常见环境、激活环境：
   conda create -n STMTrack python=3.7 -y
conda activate STMTrack
3. 安装torch 和torchvision

   pip install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.0 
   
   
   # pytorch v1.5.0, v1.6.0, or higher should also be OK. 
   
   
   

   gayhub上面用的是conda，但是我用conda总是出问题，所以使用pip安装
   安装不了试一下其他的源，在pip的结尾加上 :
   -i https://pypi.douban.com/simple/ #豆瓣源
4. 安装其他要求的软件
   如果在Windows上面运行会有一些麻烦，最好在Linux上面搞
   pip install -r requirements.txt
5. 下载作者训练好的模型
   在got上训练的https://drive.google.com/file/d/1AT6SAieig8oNQ-MJ6dUhCfgYCyJEdxfj/view

在全部数据集上训练的https://drive.google.com/file/d/1w7nhGZR53FQnh3fVbIcbj08hxa2Zjvub/view

5. 其他配置
   · 在STMTrack-main/experiments/stmtrack/test/目录下对otb，uav，got等数据集进行配置，以uav为例
   · 打开STMTrack/experiments/stmtrack/test/UAV123/stmtrack-googlenet-uav123.yaml，然后在yaml文件的最后一行添加数据集的路径 data_root: "数据集的绝对路径"，只把引号里面的内容改成路径即可，引号不要删
   · 继续修改与训练模型的位置，也是上面的那个yaml文件，大概在第33行，或者直接搜索pretain_model_path:即可，pretrain_model_path: "/root/STMTrack/epoch-19_fulldata.pkl" ，双引号里面是我的路径，改成你们自己的即可，提醒一下，要看一下原来文件中的与训练文件写的是fulldata.pk1还是gotdata.pk1,改成对应文件的路径
   · 下载uav123.json和lasot.json，gayhub里面没有这个，这个在analyst的原工程文件里面有，我把链接放到下面，大家可以使用git clone克隆完再复制进去。git clone https://github.com/megvii-research/video_analyst/tree/master/videoanalyst/evaluation/got_benchmark/datasets ,STMTrack是把analyst放在STMTrack文件夹里面的，进去即可看到，后面的e路径是一样的。

这样基本上就完成了，可以进行测试了,首先如STMTrack这个目录，然后在终端输入python main/test.py --config /root/STMTrack/experiments/stmtrack/test/UAV123/stmtrack-googlenet-uav123.yaml，一些刚入门的同学可能会忘记输入--config /....这些内容，如果是使用命令行这个是一定要加上的，否则没有对应的参数是无法运行成功的。如果是使用pycharm则是要在test.py文件中把相应的参数改成要用的实际参数。

如果是运行otb数据集，则需要改一下otb.py文件中的内容。目录是：/root/STMTrack/videoanalyst/evaluation/got_benchmark/datasets/otb.py，这个文件中的第43行，46行，55行需要改一些代码，把human4改成human4-2,把Skating2改成Skating2-1，然后在后面再添加一个Skating2-2，把Jogging改成Jogging-1,然后添加一个Jogging-2。
原因和数据集可在看我的这边文章 otb数据集的一些问题