From 810476e6ca8be9041986c77f891f372e5c9d5381 Mon Sep 17 00:00:00 2001 From: LokeZhou Date: Fri, 9 Dec 2022 14:59:36 +0800 Subject: [PATCH] mot_sde_infer.py fix use ocsort_tracker bug (#7482) --- deploy/pptracking/python/mot_sde_infer.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/deploy/pptracking/python/mot_sde_infer.py b/deploy/pptracking/python/mot_sde_infer.py index d8787ec70..f7dca28e4 100644 --- a/deploy/pptracking/python/mot_sde_infer.py +++ b/deploy/pptracking/python/mot_sde_infer.py @@ -654,6 +654,11 @@ class SDE_Detector(Detector): fps = 1. / timer.duration if self.use_deepsort_tracker or self.use_ocsort_tracker: # 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( (frame_id + 1, online_tlwhs, online_scores, online_ids)) im = plot_tracking( -- GitLab