Update extract_frames.py

extract_frames.py

@@ -19,16 +19,13 @@ def preprocess_and_save_frames(
     device: str = "cpu"
 ):
     """
-    Loads a video and saves frames as images in output_root/<video_id>/.
+    Loads a video and saves frames as images in output_root/.
+    Frame names follow the pattern: <video_id>_frame_<n>.png
     """
 
     # Ensure the base frames directory exists
     os.makedirs(output_root, exist_ok=True)
 
-    # Create subfolder for this specific video
-    video_dir = os.path.join(output_root, video_id)
-    os.makedirs(video_dir, exist_ok=True)
-
     center_crop_transform = None
     if crop_size > 0:
         from torchvision import transforms
@@ -47,9 +44,14 @@ def preprocess_and_save_frames(
             if center_crop_transform is not None:
                 frame_tensor = center_crop_transform(frame_tensor)
 
-            # Updated naming pattern
-            frame_path = os.path.join(video_dir, f"_frame_{frame_count}.png")
+            # Flat naming pattern (no subfolders)
+            frame_name = f"{video_id}_frame_{frame_count}.png"
+            frame_path = os.path.join(output_root, frame_name)
             save_image(frame_tensor, frame_path)
+
+            # Print saved frame name
+            print(frame_name)
+
             frame_count += 1
 
             frame_bytes = frame_tensor.numel() * 4
@@ -74,4 +76,4 @@ if __name__ == "__main__":
         except Exception as e:
             print(f"⚠️ Failed {el['id']}: {e}")
 
-    print("✅ All frames saved under ./frames/<video_id>/")
+    print("✅ All frames saved under ./frames/")