Update datasets.py (#3216)

2021-05-17 22:24:26 +02:00 · 2021-05-17 22:24:26 +02:00 · 13a1c72699
parent ffb47ffbeb
commit 13a1c72699
1 changed files with 1 additions and 1 deletions
--- a/utils/datasets.py
+++ b/utils/datasets.py
@ -284,7 +284,7 @@ class LoadStreams:  # multiple IP or RTSP cameras
            assert cap.isOpened(), f'Failed to open {s}'
            w = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
            h = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
-            self.fps = cap.get(cv2.CAP_PROP_FPS) % 100
+            self.fps = (cap.get(cv2.CAP_PROP_FPS) % 100) or 30.0  # assume 30 FPS if cap gets 0 FPS
            self.frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))

            _, self.imgs[i] = cap.read()  # guarantee first frame