LoadStreams() frame loss bug fix (#2222)

ultralytics · Feb 15, 2021 · 7b833e3 · 7b833e3
1 parent 3ff783c
commit 7b833e3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/utils/datasets.py b/utils/datasets.py
@@ -300,7 +300,8 @@ def update(self, index, cap):
             # _, self.imgs[index] = cap.read()
             cap.grab()
             if n == 4:  # read every 4th frame
-                _, self.imgs[index] = cap.retrieve()
+                success, im = cap.retrieve()
+                self.imgs[index] = im if success else self.imgs[index] * 0
                 n = 0
             time.sleep(0.01)  # wait time