gggmt commited on
Commit
1679aac
·
unverified ·
1 Parent(s): e8f8f2b

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

Browse files
Files changed (1) hide show
  1. 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
- if not ret_val:
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