glenn-jocher commited on
Commit
7b833e3
·
unverified ·
1 Parent(s): 3ff783c

LoadStreams() frame loss bug fix (#2222)

Browse files
Files changed (1) hide show
  1. utils/datasets.py +2 -1
utils/datasets.py CHANGED
@@ -300,7 +300,8 @@ class LoadStreams: # multiple IP or RTSP cameras
300
  # _, self.imgs[index] = cap.read()
301
  cap.grab()
302
  if n == 4: # read every 4th frame
303
- _, self.imgs[index] = cap.retrieve()
 
304
  n = 0
305
  time.sleep(0.01) # wait time
306
 
 
300
  # _, self.imgs[index] = cap.read()
301
  cap.grab()
302
  if n == 4: # read every 4th frame
303
+ success, im = cap.retrieve()
304
+ self.imgs[index] = im if success else self.imgs[index] * 0
305
  n = 0
306
  time.sleep(0.01) # wait time
307