fix empty frame

kaylode · Feb 8, 2022 · f396a62 · f396a62
1 parent 2319714
commit f396a62
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/modules/__init__.py b/modules/__init__.py
@@ -64,6 +64,9 @@ def run(self):
                     frame_id = batch['frames'][i]
 
                     ori_img = ori_imgs[i]
+
+                    if len(boxes) == 0:
+                        continue
                     track_result = self.tracker.run(ori_img, boxes, labels, scores)