Update datasets.py (ultralytics#3216)

fhkiel-mlaip · May 17, 2021 · 617637e · 617637e
1 parent 03204bb
commit 617637e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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