使用最新目标跟踪框mmtracking实现自己的目标跟踪项目
使用最新目标跟踪框mmtracking实现自己的目标跟踪项目
之前也一直在关注目标跟踪方面的项目,尤其是多目标跟踪,目前比较主流的多目标跟踪都是基于deepsort
来做的,这方面开源的研究工作也比较少,最近突然看到Open-mmlab
开源的mmtracking
甚是欣喜,还是热乎的,准备好好研究一下!!!
mmtraking
是20210104
开源,第一版:V0.5.0
- mmtracking官方教程文档
1 搭建环境
1、创建虚拟环境
conda create -n mmtrack python=3.7 -y
2、激活虚拟环境
conda activate mmtrack
3、安装cudatoolkit
conda install cudatookit=10.1
如果安装失败,也可以去这里先下载cudatoolkit的离线包,然后再安装:
conda install cudatoolkit-10.1.243-h6bb024c_0.tar.bz2
4、安装torch和torchvision
pip install torch==1.6.0 torchvision==0.7.0
注意:
torch和torchvision也是存在版本对应关系的!
5、安装mmcv-full
、mmdet
pip install mmcv-full mmdet
如果安装失败,也可以去pypi
官网先下载编译好的whl
安装包,然后再安装
注意:
mmcv-full 和mmdet
存在版本依赖问题,可能会出现版本不对应的错误,因此我这里下载的都是最新版本的,mmcv-full=1.2.5、mmdet=2.8.0
(安装时间20210106)
6、克隆仓库代码
git clone https://github.com/open-mmlab/mmtracking.git
cd mmtracking
7、安装和编译依赖
pip install -r requirements/build.txt
python setup.py develop
编译成功打印输出信息如下:
(mmtrack) shl@zhihui-mint:~/shl_res/1_project/mmtracking$ python setup.py develop
running develop
running egg_info
writing mmtrack.egg-info/PKG-INFO
writing dependency_links to mmtrack.egg-info/dependency_links.txt
writing requirements to mmtrack.egg-info/requires.txt
writing top-level names to mmtrack.egg-info/top_level.txt
reading manifest file 'mmtrack.egg-info/SOURCES.txt'
writing manifest file 'mmtrack.egg-info/SOURCES.txt'
running build_ext
Creating /home/shl/anaconda3/envs/mmtrack/lib/python3.7/site-packages/mmtrack.egg-link (link to .)
mmtrack 0.5.0 is already the active version in easy-install.pth
Installed /home/shl/shl_res/1_project/mmtracking
Processing dependencies for mmtrack==0.5.0
Searching for flake8
Reading https://pypi.org/simple/flake8/
Downloading https://files.pythonhosted.org/packages/d4/ca/3971802ee6251da1abead1a22831d7f4743781e2f743bd266bdd2f46c19b/flake8-3.8.4-py2.py3-none-any.whl#sha256=749dbbd6bfd0cf1318af27bf97a14e28e5ff548ef8e5b1566ccfb25a11e7c839
Best match: flake8 3.8.4
Processing flake8-3.8.4-py2.py3-none-any.whl
Installing flake8-3.8.4-py2.py3-none-any.whl to /home/shl/anaconda3/envs/mmtrack/lib/python3.7/site-packages
Adding flake8 3.8.4 to easy-install.pth file
Installing flake8 script to /home/shl/anaconda3/envs/mmtrack/bin
Installed /home/shl/anaconda3/envs/mmtrack/lib/python3.7/site-packages/flake8-3.8.4-py3.7.egg
Searching for setuptools_scm
Reading https://pypi.org/simple/setuptools_scm/
Downloading https://files.pythonhosted.org/packages/db/6e/2815f7c8561b088ccedc128681e64daac3d6b2e81a9918b007e244dad8b1/setuptools_scm-5.0.1-py2.py3-none-any.whl#sha256=62fa535edb31ece9fa65dc9dcb3056145b8020c8c26c0ef1018aef33db95c40d
Best match: setuptools-scm 5.0.1
Processing setuptools_scm-5.0.1-py2.py3-none-any.whl
Installing setuptools_scm-5.0.1-py2.py3-none-any.whl to /home/shl/anaconda3/envs/mmtrack/lib/python3.7/site-packages
Adding setuptools-scm 5.0.1 to easy-install.pth file
Installed /home/shl/anaconda3/envs/mmtrack/lib/python3.7/site-packages/setuptools_scm-5.0.1-py3.7.egg
Searching for pytz>=2017.3
Reading https://pypi.org