multimodalart HF Staff commited on
Commit
71a657e
·
1 Parent(s): b9ae405

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +3 -0
utils.py CHANGED
@@ -31,6 +31,9 @@ def save_video_frames(video_path, img_size=(512,512)):
31
  def video_to_frames(video_path, img_size=(512,512)):
32
  video, _, video_info = read_video(video_path, output_format="TCHW")
33
  print(video_info["fps"])
 
 
 
34
  # rotate video -90 degree if video is .mov format. this is a weird bug in torchvision
35
  if video_path.endswith('.mov'):
36
  video = T.functional.rotate(video, -90)
 
31
  def video_to_frames(video_path, img_size=(512,512)):
32
  video, _, video_info = read_video(video_path, output_format="TCHW")
33
  print(video_info["fps"])
34
+ print(video_info["video_fps"])
35
+ print(video_info)
36
+
37
  # rotate video -90 degree if video is .mov format. this is a weird bug in torchvision
38
  if video_path.endswith('.mov'):
39
  video = T.functional.rotate(video, -90)