Update extract_frames.py

extract_frames.py

@@ -37,6 +37,7 @@ def preprocess_and_save_frames(
     file_like.seek(0)
     container = av.open(file_like)
     current_memory = 0
+    frame_count = 1  # Start counting frames at 1
 
     for i, frame in enumerate(container.decode(video=0)):
         if i % every == 0:
@@ -46,8 +47,10 @@ def preprocess_and_save_frames(
             if center_crop_transform is not None:
                 frame_tensor = center_crop_transform(frame_tensor)
 
-            frame_path = os.path.join(video_dir, f"frame_{i:05d}.png")
+            # Updated naming pattern
+            frame_path = os.path.join(video_dir, f"_frame_{frame_count}.png")
             save_image(frame_tensor, frame_path)
+            frame_count += 1
 
             frame_bytes = frame_tensor.numel() * 4
             current_memory += frame_bytes