SiamMask目標跟蹤算法項目環境配置記錄

一、SiamMask

2019目標跟蹤算法新高度,記錄一下環境配置過程,感謝大神的代碼開源。Fast Online Object Tracking and Segmentation: A Unifying Approach ,論文地址【Paper】

二、運行環境配置

原運行環境是在Ubuntu 16.04, Python 3.6, Pytorch 0.4.1, CUDA 9.2, RTX 2080 GPUs上運行的,實測的時候我的環境是Ubuntu 16.04, Python 3.7, Pytorch 0.4.1, i7的CPU

1、下載項目

git clone https://github.com/foolwood/SiamMask.git && cd SiamMask
export SiamMask=$PWD

2、設置環境

conda create -n SiamMask python=3.7
source activate SiamMask
pip install -r requirements.txt
bash make.sh

附:一般安裝的ubantu會有兩個版本的python,python2和python3,而默認的python2的版本,如何切換版本參考鏈接【參考鏈接】,以及一個可能出現的問題鏈接【解決鏈接】

3、下載模型

cd $SiamMask/experiments/siammask_sharp
wget http://www.robots.ox.ac.uk/~qwang/SiamMask_VOT.pth
wget http://www.robots.ox.ac.uk/~qwang/SiamMask_DAVIS.pth

4、運行測試demo

cd $SiamMask/experiments/siammask_sharp
export PYTHONPATH=$PWD:$PYTHONPATH
python ../../tools/demo.py --resume SiamMask_DAVIS.pth --config config_davis.json

假如出現這樣的報錯:File "../../tools/demo.py", line 7, in <module> from tools.test import * ModuleNotFoundError: No module named 'tools'

參考作者自己給出的答案修改:鏈接地址

5、demo展示

demo

假如需要修改其它視頻數據進行測試,只需要更改tools文件夾下的demo.py的路徑即可。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章