一、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展示
假如需要修改其它視頻數據進行測試,只需要更改tools文件夾下的demo.py的路徑即可。