From 0a2153c9bf584086670abda5d741b904577c60bf Mon Sep 17 00:00:00 2001
From: George Ni <nemonameless@qq.com>
Date: Fri, 30 Jul 2021 18:57:00 +0800
Subject: [PATCH] [mot] fix mot pose unite deploy cpu mkldnn (#3837)

---
 deploy/python/mot_jde_infer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/deploy/python/mot_jde_infer.py b/deploy/python/mot_jde_infer.py
index d35a39629..584fcb384 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:
-- 
GitLab