fcakyon commited on
Commit
7b5f593
1 Parent(s): 66ccdd2

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -22,7 +22,7 @@ def sample_frames_from_video_file(
22
  # sample frames
23
  start_idx = 0
24
  end_idx = num_frames * frame_sampling_rate - 1
25
- indices = np.linspace(start_idx, end_idx, num=num_frames, dtype=int)
26
  frames = videoreader.get_batch(indices).asnumpy()
27
 
28
  return frames
@@ -35,7 +35,7 @@ def get_num_total_frames(file_path: str):
35
 
36
 
37
  def convert_frames_to_gif(frames, save_path: str = "frames.gif"):
38
- converted_frames = frames.astype(int)
39
  Path(save_path).parent.mkdir(parents=True, exist_ok=True)
40
  imageio.mimsave(save_path, converted_frames, duration=125) # 8fps
41
  return save_path
 
22
  # sample frames
23
  start_idx = 0
24
  end_idx = num_frames * frame_sampling_rate - 1
25
+ indices = np.linspace(start_idx, end_idx, num=num_frames, dtype=np.int64)
26
  frames = videoreader.get_batch(indices).asnumpy()
27
 
28
  return frames
 
35
 
36
 
37
  def convert_frames_to_gif(frames, save_path: str = "frames.gif"):
38
+ converted_frames = frames.astype(np.uint8)
39
  Path(save_path).parent.mkdir(parents=True, exist_ok=True)
40
  imageio.mimsave(save_path, converted_frames, duration=125) # 8fps
41
  return save_path