天天看点

MDNet视频目标跟踪源码运行笔记(Python 3.6版本)——Testing模式1. 论文基本信息2. 运行环境介绍3. 准备4. 运行5. 可能出现的问题及解决方法

1. 论文基本信息

  • 论文标题:Learning Multi-Domain Convolutional Neural Networks for Visual Tracking
  • 论文作者:Hyeonseob Nam(Dept. of Computer Science and Engineering, POSTECH, Korea)等人
  • 论文出处:CVPR 2016
  • 在线阅读:https://arxiv.org/pdf/1510.07945v2.pdf
  • 源码链接1:https://github.com/HyeonseobNam/MDNet (MATLAB版本)
  • 源码链接2:https://github.com/HyeonseobNam/py-MDNet (Python 3.6版本)

2. 运行环境介绍

  • NVIDIA GTX 1070
  • Windows 10 x64
  • CUDA 8.0.61 for Windows 10
  • Python 3.6
  • PyTorch 1.0.0

3. 准备

下载MDNet官方Python源码并解压。

4. 运行

进入到源码的根目录中,在终端中运行如下命令即可运行MDNet跟踪算法(请确保当前的Python版本为3.6):

python tracking/run_tracker.py -s DragonBaby -d
           

大约等待几分钟,算法会开始跟踪,这里配置的是源码包自带的视频DragonBaby。运行效果如下图所示(绿色矩形框表示ground-truth,红色矩形框表示算法result):

MDNet视频目标跟踪源码运行笔记(Python 3.6版本)——Testing模式1. 论文基本信息2. 运行环境介绍3. 准备4. 运行5. 可能出现的问题及解决方法

同时终端中会输出如下信息:

MDNet视频目标跟踪源码运行笔记(Python 3.6版本)——Testing模式1. 论文基本信息2. 运行环境介绍3. 准备4. 运行5. 可能出现的问题及解决方法

5. 可能出现的问题及解决方法

问题 1: 找不到numpy模块,

ModuleNotFoundError: No module named 'numpy'

解决: 在自己的Python 3.6环境中安装numpy库,可以用命令

conda install numpy

进行安装。

问题 2: 找不到yaml模块,

ModuleNotFoundError: No module named 'yaml'

解决: 在自己的Python 3.6环境中安装yaml库,可以用命令

pip install pyyaml

进行安装。

问题 3: 找不到PIL模块,

ModuleNotFoundError: No module named 'PIL'

解决: 在自己的Python 3.6环境中安装Pillow库,可以用命令

pip install Pillow

进行安装。

问题 4: 找不到matplotlib模块,

ModuleNotFoundError: No module named 'matplotlib'

解决: 在自己的Python 3.6环境中安装matplotlib库,可以用命令

conda install matplotlib

进行安装。

问题 5: 找不到scipy模块,

ModuleNotFoundError: No module named 'scipy'

解决: 在自己的Python 3.6环境中安装scipy库,可以用命令

conda install scipy

进行安装。

问题 6: 在安装了scipy后报错:

from scipy.misc import imresize
ImportError: cannot import name 'imresize'
           

解决: 换一种写法,通过numpy来实现,原理请参考:https://blog.csdn.net/discoverer100/article/details/95534621 。

修改源码,在

modules/util.py

文件中,将第1行换为:

from PIL import Image
           

将该文件的第132行的imresize语句改为:

问题 7: 找不到opencv-python模块,

ModuleNotFoundError: No module named 'cv2'

解决: 在自己的Python 3.6环境中安装opencv-python库,可以用命令

pip install opencv-python

进行安装。

问题 8: 找不到sklearn模块,

ImportError: No module named sklearn.linear_model

解决: 在自己的Python 3.6环境中安装scikit-learn库,可以用命令

conda install scikit-learn

进行安装。

继续阅读