| #!/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) | |