From c76c8c2748256d072ee9bdbfdb071b4eccf35d05 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 17 May 2021 22:16:33 +0200 Subject: [PATCH] Update datasets.py --- utils/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/datasets.py b/utils/datasets.py index 56db9d2f13b5..a14099875455 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -284,7 +284,7 @@ def __init__(self, sources='streams.txt', img_size=640, stride=32): 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