使用最新目标跟踪框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/simple/pytz/ Downloading https