Update `LoadImages` `ret_val=False` handling (#5852)

Video errors may occur.
2021-12-02 17:57:39 +08:00 · 2021-12-02 17:57:39 +08:00 · 1679aacdc7
parent e8f8f2b903
commit 1679aacdc7
1 changed files with 1 additions and 1 deletions
--- a/utils/datasets.py
+++ b/utils/datasets.py
@ -200,7 +200,7 @@ class LoadImages:
            # 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