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):
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语句改为:
scaled
=
np
.
array
(
Image
.
fromarray
(
cropped
)
.
resize
(
img_size
,
img_size
)
)
问题 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
进行安装。