Upload extract_frames.py

scripts/extract_frames.py

@@ -9,15 +9,22 @@ def extract_frames(video_path, output_dir, fps=2):
     count = 0
     frame_id = 1
     success, image = vidcap.read()
+    failed = 0
     while success:
-        if int(vidcap.get(cv2.CAP_PROP_POS_FRAMES)) % int(vidcap.get(cv2.CAP_PROP_FPS) // fps) == 0:
+        if int(vidcap.get(cv2.CAP_PROP_POS_FRAMES)) % max(1, int(vidcap.get(cv2.CAP_PROP_FPS) // fps)) == 0:
             frame_name = f"{video_name}_{frame_id:04d}.png"
-            cv2.imwrite(os.path.join(output_dir, frame_name), image)
+            out_path = os.path.abspath(os.path.join(output_dir, frame_name))
+            ok = cv2.imwrite(out_path, image)
+            if not ok:
+                print(f"[ERROR] Failed to write frame: {out_path}")
+                failed += 1
+            elif frame_id <= 5:
+                print(f"[DEBUG] Wrote frame: {out_path}")
             frame_id += 1
         success, image = vidcap.read()
         count += 1
     vidcap.release()
-    print(f"Extracted {frame_id-1} frames from {video_path}")
+    print(f"Extracted {frame_id-1} frames from {video_path}. Failed writes: {failed}")
 
 if __name__ == "__main__":
     video_dir = "videos"