diff --git a/deploy/python/mot_jde_infer.py b/deploy/python/mot_jde_infer.py index d35a396298b52f3e6b9b3b49b7548c2280c791b3..584fcb384e65d2aa1a59025f2f3f522fe75262e8 100644 --- a/deploy/python/mot_jde_infer.py +++ b/deploy/python/mot_jde_infer.py @@ -91,6 +91,8 @@ class JDE_Detector(Detector): def postprocess(self, pred_dets, pred_embs, threshold): online_targets = self.tracker.update(pred_dets, pred_embs) + if online_targets == []: + return [pred_dets[0][:4]], [pred_dets[0][4]], [1] online_tlwhs, online_ids = [], [] online_scores = [] for t in online_targets: