Spaces:
Runtime error
Runtime error
Commit
·
a2c56f0
1
Parent(s):
20ea6da
Update icpr2020dfdc/blazeface/read_video.py
Browse files
icpr2020dfdc/blazeface/read_video.py
CHANGED
@@ -37,7 +37,7 @@ class VideoReader:
|
|
37 |
frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
|
38 |
if frame_count <= 0: return None
|
39 |
|
40 |
-
frame_idxs = np.linspace(0, frame_count - 1, num_frames, endpoint=True, dtype=np.
|
41 |
frame_idxs = np.unique(frame_idxs) # Avoid repeating frame idxs otherwise it breaks reading
|
42 |
if jitter > 0:
|
43 |
np.random.seed(seed)
|
|
|
37 |
frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
|
38 |
if frame_count <= 0: return None
|
39 |
|
40 |
+
frame_idxs = np.linspace(0, frame_count - 1, num_frames, endpoint=True, dtype=np.int64)
|
41 |
frame_idxs = np.unique(frame_idxs) # Avoid repeating frame idxs otherwise it breaks reading
|
42 |
if jitter > 0:
|
43 |
np.random.seed(seed)
|