不得不說王強大神是真的強,如果過幾年能去中科院,或許能有機會和人家一起工作嘻嘻嘻。但還是先別想屁喫,好好學習纔是正經。配了兩天環境之後,今天終於跑起了siammask的demo。雖說官方github項目上有詳細步驟,但我畢竟linux系統不熟,還是廢了一番力氣才成功跑通……看評論也有人遇到了和我一樣的問題,特此記錄。
項目代碼
運行環境
This code has been tested on Ubuntu 16.04, Python 3.6, Pytorch 0.4.1, CUDA 9.2, RTX 2080 GPUs
上面的是原項目的運行環境,我的運行環境是ubuntu18.04, CUDA10.1, GTX1060。其他軟件和原項目一樣。
Clone the repository
git clone https://github.com/foolwood/SiamMask.git && cd SiamMask
export SiamMask=$PWD
$PWD是linux系統下的命令,返回值是當前路徑,所以上面的export語句必須在cd Siammask的情況下執行。
Setup python environment
conda create -n siammask python=3.6
source activate siammask
pip install -r requirements.txt
bash make.sh
Add the project to your PYTHONPATH
export PYTHONPATH=$PWD:$PYTHONPATH
這裏因爲文件夾沒有變過,所以$PWD返回值還是當前路徑,故上面語句的作用是把Siammask所在文件夾加入python路徑。
Demo
Setup your environment & Download the SiamMask model
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
Run demo.py
cd $SiamMask/experiments/siammask_sharp
export PYTHONPATH=$PWD:$PYTHONPATH
python ../../tools/demo.py --resume SiamMask_DAVIS.pth --config config_davis.json
這裏可能會遇到如下錯誤:
run python ../../tools/demo.py --resume SiamMask_DAVIS.pth --config config_davis.json and get a error:
Traceback (most recent call last):
File "../tools/demo.py", line 7, in
from tools.test import *
ModuleNotFoundError: No module named 'tools'
首先:
ROOT=`git rev-parse --show-toplevel`
export PYTHONPATH=$ROOT:$PYTHONPATH
然後,注意我們現在應該是處於SiamMask/experiments/siammask_sharp
文件路徑之下,在這個環境下運行:
export PYTHONPATH=$PWD:$PYTHONPATH
我們就成功的siammask_sharp文件夾加入了python路徑。之後再運行demo,就可以成功了。注意路徑的聲明是一次性的,只在當前終端打開的時候有效。