Update preprocessing.py

preprocessing.py

@@ -57,6 +57,7 @@ def pad_and_resize(frames, target_size):
     ])
     processed_frames = [transform(frame) for frame in frames]
     processed_frames = torch.stack(processed_frames)
-    # return processed_frames.permute(1, 0, 2, 3).unsqueeze(0)  # Add batch dimension and permute
-    return processed_frames.permute(0, 2, 3, 1).unsqueeze(0)
+    # return processed_frames.permute(1, 0, 2, 3).unsqueeze(0)  # Add batch dimension and permute [3, 24, 224, 224]
+    # return processed_frames.permute(0, 2, 3, 1).unsqueeze(0) # [24, 224, 224, 3]
+    return processed_frames.permute(0, 1, 2, 3).unsqueeze(0)