01mmtracking多目标跟踪算法 deepsort 复现

Posted 2022-01-28 计算机视觉-杨帆

github：https://github.com/open-mmlab/mmtracking
码云（新版）：https://gitee.com/YFwinston/mmdetection.git
论文地址：Simple online and realtime tracking with a deep association metric
GPU平台：极链AI
B站视频：https://www.bilibili.com/video/BV1Au411U7y4?spm_id_from=333.999.0.0

本系列的链接

00【mmtracking】mmtracking 目标跟踪 介绍

01【mmtracking】多目标跟踪算法 deepsort 复现

目录

• 本系列的链接
• 1 项目下载与安装
• • 1.1 AI平台
  • 1.2 安装mmtracking
• 2 demo测试与结果
• 3 运行中的错误解决

1 项目下载与安装

1.1 AI平台

我还是选择的极链AI云平台
下面是创建实例镜像的pytorch、python、cuda的版本，pytorch=1.8.0，python=3.8，cuda=11.1.1

1.2 安装mmtracking

创建好镜像之后，在终端输入下面的命令

cd home

pip install mmcv-full==1.3.17 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html

pip install opencv-python-headless==4.1.2.30

git clone https://gitee.com/YFwinston/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e .
cd ..

git clone https://gitee.com/YFwinston/mmtracking.git
cd mmtracking
pip install -r requirements/build.txt
pip install -v -e .

pip install git+https://gitee.com/YFwinston/toolkit.git

2 demo测试与结果

python demo/demo_mot_vis.py  configs/mot/deepsort/sort_faster-rcnn_fpn_4e_mot17-private.py  --input demo/demo.mp4  --output mot.mp4 

最后结果

3 运行中的错误解决

如果在运行中出现一下错误：

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-1-c8ec22b3e787> in <module>
----> 1 import cv2

/opt/conda/lib/python3.8/site-packages/cv2/__init__.py in <module>
      7 
      8 from .cv2 import *
----> 9 from .cv2 import _registerMatType
     10 from . import mat_wrapper
     11 from . import gapi

ImportError: cannot import name '_registerMatType' from 'cv2.cv2' (/opt/conda/lib/python3.8/site-packages/cv2/cv2.cpython-38-x86_64-linux-gnu.so)

解决方案如下：
在终端运行：

pip install opencv-python-headless==4.1.2.30