From e25a07785180b5e491aa9b3d8ee0b79e9ef82567 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 15 Feb 2021 11:02:20 -0800
Subject: [PATCH] LoadStreams() frame loss bug fix (#2222)

---
 utils/datasets.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/utils/datasets.py b/utils/datasets.py
index 29a8812a20a2..4f2939d4bef2 100755
--- 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