当前位置: 首页 > 工具软件 > MMTracking > 使用案例 >

MMTracking-1.0.0rc1单目标跟踪算法之单目标或将多目标按单目标预测的轨迹可视化

蓟俊杰
2023-12-01

阅读了b站大佬同济子豪兄有关MMTracking轨迹可视化的相关代码,但其代码在被MMengine统一后的MMTracking-1.0.0rc1以上版本不太适用,故作此文。

  • 我们对原始代码做了必要的修改,首先是用SOT算法对单目标的跟踪做轨迹可视化:
import mmcv
import mmengine
import tempfile
import cv2
import os.path as osp
from mmtrack.registry import VISUALIZERS
from mmtrack.apis import inference_sot, init_model
from mmtrack.utils import register_all_modules

register_all_modules(init_default_scope=True) # 没有这一行所有模型都会报错

# 输入输出视频路径
 类似资料: