From c8de4fe63486a7e153d6c26524503a740e486d1a Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 15 Sep 2023 02:35:22 +0200
Subject: [PATCH] Fix LoadStreams missing frame bug (#4913)

---
 ultralytics/data/loaders.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/ultralytics/data/loaders.py b/ultralytics/data/loaders.py
index 8524f6d5..462d70d8 100644
--- a/ultralytics/data/loaders.py
+++ b/ultralytics/data/loaders.py
@@ -127,8 +127,10 @@ class LoadStreams:
                 if not self.threads[i].is_alive() or cv2.waitKey(1) == ord('q'):  # q to quit
                     self.close()
                     raise StopIteration
-                LOGGER.warning(f'WARNING ⚠️ Waiting for stream {i}')
                 time.sleep(1 / min(self.fps))
+                x = self.imgs[i]
+                if not x:
+                    LOGGER.warning(f'WARNING ⚠️ Waiting for stream {i}')
 
             # Get and remove the first frame from imgs buffer
             if self.buffer: