LoadStreams() frame loss bug fix (#2222)
Browse files- 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 |
-
|
|
|
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 |
|