add keyframe, a tool to extract keyframes from videos with ffmpeg

keyframe

@@ -0,0 +1,42 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import os
+import sys
+import subprocess
+
+
+def extract_keyframes(video_file_path):
+    """
+    Extract keyframes from the given video file.
+
+    Parameters:
+    video_file_path (str): The path to the video file.
+    """
+    # Get the base name of the video file without extension
+    base_name = os.path.splitext(os.path.basename(video_file_path))[0]
+
+    # Create a directory with the same name as the video file
+    output_dir = os.path.join(os.path.dirname(video_file_path), base_name)
+    os.makedirs(output_dir, exist_ok=True)
+
+    # Command to extract keyframes using ffmpeg
+    ffmpeg_command = [
+        'ffmpeg',
+        '-i', video_file_path,
+        '-vf', 'select=eq(pict_type\\,I)',
+        '-vsync', 'vfr',
+        os.path.join(output_dir, 'frame_%03d.png')
+    ]
+
+    # Run the ffmpeg command
+    subprocess.run(ffmpeg_command, check=True)
+
+
+if __name__ == "__main__":
+    if len(sys.argv) != 2:
+        print("Usage: python keyframe.py <path_to_video>")
+        sys.exit(1)
+
+    video_path = sys.argv[1]
+    extract_keyframes(video_path)