From 6f82270a8b77df1a9138ea4eb29f116ea83515e1 Mon Sep 17 00:00:00 2001
From: gggmt <1065504814@qq.com>
Date: Thu, 2 Dec 2021 15:54:00 +0800
Subject: [PATCH] Update datasets.py

Video errors may occur.
---
 utils/datasets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/datasets.py b/utils/datasets.py
index f3abfb1f6f90..6ce7a81b69e7 100755
--- 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