Siammask源碼demo運行配置

不得不說王強大神是真的強,如果過幾年能去中科院,或許能有機會和人家一起工作嘻嘻嘻。但還是先別想屁喫,好好學習纔是正經。配了兩天環境之後,今天終於跑起了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,就可以成功了。注意路徑的聲明是一次性的,只在當前終端打開的時候有效。

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