malmukhtar commited on
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.int)
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)