Merge pull request #17 from kaylode/fix

fix empty frame
kaylode · Feb 8, 2022 · 5007a54 · 5007a54
2 parents 2319714 + f396a62
commit 5007a54
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)