gggmt
commited on
Update `LoadImages` `ret_val=False` handling (#5852)
Browse files- utils/datasets.py +1 -1
utils/datasets.py
CHANGED
@@ -200,7 +200,7 @@ class LoadImages:
|
|
200 |
# Read video
|
201 |
self.mode = 'video'
|
202 |
ret_val, img0 = self.cap.read()
|
203 |
-
|
204 |
self.count += 1
|
205 |
self.cap.release()
|
206 |
if self.count == self.nf: # last video
|
|
|
200 |
# Read video
|
201 |
self.mode = 'video'
|
202 |
ret_val, img0 = self.cap.read()
|
203 |
+
while not ret_val:
|
204 |
self.count += 1
|
205 |
self.cap.release()
|
206 |
if self.count == self.nf: # last video
|