下載、安裝與使用 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” 一行去掉即可

點擊運行，我們就可以看到效果了