未验证 提交 810476e6 编写于 作者: L LokeZhou 提交者: GitHub

mot_sde_infer.py fix use ocsort_tracker bug (#7482)

上级 45ba40b4
...@@ -654,6 +654,11 @@ class SDE_Detector(Detector): ...@@ -654,6 +654,11 @@ class SDE_Detector(Detector):
fps = 1. / timer.duration fps = 1. / timer.duration
if self.use_deepsort_tracker or self.use_ocsort_tracker: if self.use_deepsort_tracker or self.use_ocsort_tracker:
# use DeepSORTTracker or OCSORTTracker, only support singe class # use DeepSORTTracker or OCSORTTracker, only support singe class
if isinstance(online_tlwhs, defaultdict):
online_tlwhs = online_tlwhs[0]
online_scores = online_scores[0]
online_ids = online_ids[0]
results[0].append( results[0].append(
(frame_id + 1, online_tlwhs, online_scores, online_ids)) (frame_id + 1, online_tlwhs, online_scores, online_ids))
im = plot_tracking( im = plot_tracking(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册