Update datasets.py

Video errors may occur.
ultralytics · Dec 2, 2021 · 6f82270 · 6f82270
1 parent e8f8f2b
commit 6f82270
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/datasets.py b/utils/datasets.py
@@ -200,7 +200,7 @@ def __next__(self):
             # Read video
             self.mode = 'video'
             ret_val, img0 = self.cap.read()
-            if not ret_val:
+            while not ret_val:
                 self.count += 1
                 self.cap.release()
                 if self.count == self.nf:  # last video