Update utils.py

padding and aspect ratio changes

utils.py

@@ -40,6 +40,15 @@ def video_to_frames(video_path, img_size=(512,512)):
     for i in range(len(video)):
         ind = str(i).zfill(5)
         image = T.ToPILImage()(video[i])
+        
+        # get new height and width to maintain aspect ratio
+        height, width = image.shape
+        new_height = img_size[0] * height / width 
+        new_width  = img_size[1] * width / height
+        
+        # pad
+        image = Image.new(image.mode, (new_width, new_height), (0, 0, 0))
+        
         image_resized = image.resize((img_size),  resample=Image.Resampling.LANCZOS)
         # image_resized.save(f'data/{video_name}/{ind}.png')
         frames.append(image_resized)