LoadStreams() frame loss bug fix (ultralytics#2222)

KMint1819 · Feb 15, 2021 · e25a077 · e25a077
1 parent 083e5f1
commit e25a077
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