Update utils.py

utils.py

@@ -30,9 +30,6 @@ def save_video_frames(video_path, img_size=(512,512)):
         
 def video_to_frames(video_path, img_size=(512,512)):
     video, _, video_info = read_video(video_path, output_format="TCHW")
-    print(video_info)
-    print(video_info["video_fps"])
-    
     
     # rotate video -90 degree if video is .mov format. this is a weird bug in torchvision
     if video_path.endswith('.mov'):
@@ -46,7 +43,7 @@ def video_to_frames(video_path, img_size=(512,512)):
         image_resized = image.resize((img_size),  resample=Image.Resampling.LANCZOS)
         # image_resized.save(f'data/{video_name}/{ind}.png')
         frames.append(image_resized)
-    return frames
+    return frames, video_info["video_fps"]
 
 def add_dict_to_yaml_file(file_path, key, value):
     data = {}