diff --git a/deploy/pptracking/python/mot_sde_infer.py b/deploy/pptracking/python/mot_sde_infer.py index d8787ec70efab564ce296f57b82f3933f417d293..f7dca28e484a85bf4eab583d2fd40d5ca4e64f7a 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(