之前在使用SiamMask进行目标跟踪时,发现了一个名为SiamMask_e的改进版本,于是有了如下的使用总结:
在使用SiamMask_e之前,需要先安装PySOT这个项目,SiamMask_e只是这个项目的添加包(https://github.com/STVIR/pysot)
先安装pysot:
conda create --name pysot python=3.7
conda activate pysot(source activate pysot)
conda install numpy
#conda install pytorch=0.4.1 torchvision cuda90 -c pytorch(根据服务器版本更换)
conda install pytorch==0.4.1 torchvision cudatoolkit=10.0 -c pytorch
pip install opencv-python
pip install pyyaml yacs tqdm colorama matplotlib cython tensorboardX
cd /home/mfx/xmf/pysot-master
python setup.py build_ext --inplace
安装后,使用如下:
conda activate pysot(source activate pysot)
cd /home/mfx/xmf/pysot-master
export PYTHONPATH=/home/mfx/xmf/pysot-master:$PYTHONPATH
python tools/demo.py \
--config experiments/siammask_r50_l3/config.yaml \
--snapshot experiments/siammask_r50_l3/model.pth \
--video demo/bag.avi
现在开始合并siammask_e:
cd /home/mfx/xmf/pysot-master/siammask_e-master
bash install.sh /home/mfx/xmf/pysot-master
然后把/home/mfx/xmf/pysot-master/siammask_e-master/experiments/siammaske_r50_l3r 目录下的config.yaml覆盖掉 experiments/siammask_r50_l3/config.yaml即可
合并成功了
现在运行:
python tools/demo.py \
--config experiments/siammask_r50_l3/config.yaml \
--snapshot experiments/siammask_r50_l3/model.pth \
--video demo/bag.avi
则是改进版本