MMTracking-1.0.0rc1多目标跟踪算法之多目标轨迹可视化

B站大佬同济子豪兄关于多目标轨迹可视化的代码在新版本MMTracking上不太适用,一些API和数据结构都不太一样,因此代码改动比较多。下面是博主自己的可视化效果。

# Copyright (c) OpenMMLab. All rights reserved.

import mmcv
import mmengine
import tempfile
import cv2
import numpy as np
import os.path as osp
from mmtrack.registry import VISUALIZERS
from mmtrack.apis import inference_mot, init_model
from mmtrack.utils import register_all_modules
register_all_modules(init_default_scope=True) # 没有这一行所有模型都会报错
import seaborn as sns
import random

# 目标检测框线宽
thickness = 2
# 轨迹线粗细
trace_radius = 10
# 目标检测置信度阈值
score_thr = 0.0
# 置信度和 ID 单个数字的宽高
text_width = 9
te

你可能感兴趣的:(目标跟踪干货分享,轨迹绘制,mmtracking,deepsort,多目标跟踪)