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,就可以成功了。注意路径的声明是一次性的,只在当前终端打开的时候有效。

你可能感兴趣的:(计算机视觉)