下载、安装与使用 pysot

本篇文章简要介绍了如何在windows环境下，使用anaconda进行pysot运行环境的部署以及使用

pysot是商汤科技推出的一个针对单目标跟踪（Single Object Tracking）的“研究平台”，里面包含了他们团队的一些算法实现，例如SiamRPN、SiamMask等

github地址：https://github.com/STVIR/pysot
朱政对于SiamRPN系列算法的讲解视频：https://www.bilibili.com/video/BV13J411y7xP?from=search&seid=5083048807850515226

1 - 下载仓库

使用 git clone 或者下载 zip 压缩包都可以，下载代码到你的一个目录，无中文路径。例如我这里是下载的 zip 压缩包，放置的路径是 “C:\Users\SEMENTS_LAPTAP\Desktop\developArea\ObjectTrack”

解压，进入 pysot-master 文件夹

2 - 设置环境

打开 Anaconda Powershell Prompt，输入下面的命令，创建一个名为 “pysot” 的环境，指定使用 python 3.7 版本，然后切换到我们建立的这个新环境中

conda create --name pysot python=3.7
conda activate pysot

安装一些需要用到的库

conda install numpy
conda install pytorch=0.4.1 torchvision cuda90 -c pytorch
pip install opencv-python
pip install pyyaml yacs tqdm colorama matplotlib cython tensorboardX

3 - 下载模型

我们可以在 https://github.com/STVIR/pysot/blob/master/MODEL_ZOO.md 下载商汤提供的训练好的模型。他们提供了百度云下载。

下载完成后，文件夹为 PySOT，里面包含了训练好的模型，需要将其放置到我们下载的 pysot-master\experiments 文件夹中。注意：仅放置各个子文件夹的model.pth文件，切勿将百度云下载的PySOT各子文件夹中的 config.yaml 复制到 experiments中进行替换

4 - 运行 demo.py

我们使用 pycharm 打开 demo.py，点击 “File” -> “Settings” -> “Project Interpreter”，选择我们之前建立的 “pysot” 环境，没有就点击齿轮来添加

接下来需要添加 PYTONPATH 以及设置参数。

4.1 - PYTHONPATH

PYTHONPATH 以供查找包与文件关联，有两种方式可以添加指定路径到PYTHONPATH。

• 在代码文件内添加：

  import sys
  sys.path.append('你的\\pysot-master\\路径')
  

• 点击 “Run” -> “Edit Configurations”，选择左侧栏的 pyton->你要运行的代码文件。如果没有可以点击 “+” 来添加
  点击 “Environment Variables”，添加一个环境变量 “PYTHONPTH”，对应的值填写你的路径（注意 “\” 使用 “\\” 代替）

4.2 - Demo运行参数

同样点击 “Run” -> “Edit Configurations”，选择左侧栏的 pyton->你要运行的代码文件，在 “Script Path” 栏中选择 demo.py的路径。在 “Parameters” 中添加运行demo需要的一些参数。

如果你希望使用视频来进行验证，可以输入：

--config ../experiments/siamrpn_r50_l234_dwxcorr/config.yaml
--snapshot ../experiments/siamrpn_r50_l234_dwxcorr/model.pth
--video ../demo/bag.avi

这里的路径是以你在 "Script Path"栏中选定的python文件为准。这样就会使用 …/demo/bag.avi 中的视频进行；而如果你希望使用摄像头来进行验证，将 “–video …/demo/bag.avi” 一行去掉即可

点击运行，我们就可以看到效果了