使用最新目標跟蹤框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