Siam系列跟踪算法工具包PySOT配置_pysot运行

PySOT是官方出的工具包，里面包含的跟踪算法包括：

• SiamMask
• SiamRPN++
• DaSiamRPN
• SiamRPN
• SiamFC

代码下载地址：

https://github.com/STVIR/pysot

一、环境要求

语言：python

硬件：GPU

驱动：CUDA，CuDNN

深度学习：torch

辅助库：Opencv

环境的安装教程：pysot/INSTALL.md at master · STVIR/pysot · GitHub

1、确保python是否安装，python --version

2、确保驱动是否正确安装，nvidia-smi

3、确保torch和opencv正确安装

 二、运行PySOT

1、添加环境变量，将pysot工程路径加入环境变量

打开配置文件：sudo gedit ~/.bashrc

export PYTHONPATH=~/Mycode/pythoncode/pysot:$PYTHONPATH

使生效：source ~/.bashrc 

2、下载训练好的模型

下载地址：PySOT Model Zoo

在PySOT的工程路径下，新建models，然后将下载的模型放入，如下图

3、安装需要的库

pip install pyyaml yacs tqdm colorama matplotlib cython tensorboardX

4、进入工程路径

cd ~/Mycode/pythoncode/pysot/

5、运行设置文件

python setup.py build_ext --inplace

5、运行demo

python tools/demo.py --config experiments/siamrpn_r50_l234_dwxcorr/config.yaml --snapshot models/siamrpn_r50_l234_dwxcorr/model.pth

其中experiments/siamrpn_r50_l234_dwxcorr/config.yaml为配置文件所在位置，models/siamrpn_r50_l234_dwxcorr/model.pth为预训练模型所在位置。

用鼠标选择需要跟踪的目标，按C键退出程序 

6、结果