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