Python实现将MP4视频分解为JPG图片帧

pip install opencv-python

#!/usr/bin/env python3
"""
MP4视频分解为JPG图片脚本
将视频文件的每一帧保存为独立的JPG图片
"""
import cv2
import os
import argparse
from pathlib import Path
def video_to_frames(video_path, output_dir, fps=None, prefix="frame", quality=95):
    """
    将视频文件分解为JPG图片
    Args:
        video_path: 视频文件路径
        output_dir: 输出目录
        fps: 保存帧率(None表示保存所有帧，指定数字表示每秒保存多少帧)
        prefix: 输出文件名前缀
        quality: JPG图片质量(1-100)
    Returns:
        保存的图片总数
    """
    # 检查视频文件是否存在
    if not os.path.exists(video_path):
        raise FileNotFoundError(f"视频文件不存在: {video_path}")
    # 创建输出目录
    os.makedirs(output_dir, exist_ok=True)
    # 打开视频文件
    cap = cv2.VideoCapture(video_path)
    if not cap.isOpened():
        raise RuntimeError(f"无法打开视频文件: {video_path}")
    # 获取视频信息
    video_fps = cap.get(cv2.CAP_PROP_FPS)
    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
    print(f"视频信息:")
    print(f"  - 帧率: {video_fps:.2f} FPS")
    print(f"  - 总帧数: {total_frames}")
    print(f"  - 分辨率: {width}x{height}")
    print(f"  - 输出目录: {output_dir}")
    print()
    # 计算帧间隔
    if fps is not None and fps < video_fps:
        frame_interval = int(video_fps / fps)
        print(f"每隔 {frame_interval} 帧保存一次(约 {fps} FPS)")
    else:
        frame_interval = 1
        print("保存所有帧")
    # JPEG编码参数
    encode_param = [int(cv2.IMWRITE_JPEG_QUALITY), quality]
    frame_count = 0
    saved_count = 0
    try:
        while True:
            ret, frame = cap.read()
            if not ret:
                break
            # 根据帧间隔判断是否保存
            if frame_count % frame_interval == 0:
                # 生成输出文件名，使用6位数字编号
                output_path = os.path.join(output_dir, f"{prefix}_{saved_count:06d}.jpg")
                # 保存图片
                cv2.imwrite(output_path, frame, encode_param)
                saved_count += 1
                # 显示进度
                if saved_count % 100 == 0:
                    print(f"已保存 {saved_count} 张图片...")
            frame_count += 1
    finally:
        cap.release()
    print(f"\n完成! 共保存 {saved_count} 张图片到 {output_dir}")
    return saved_count
def main():
    parser = argparse.ArgumentParser(
        description="将MP4视频文件分解为JPG图片",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
示例:
  # 保存所有帧
  python video_to_frames.py input.mp4 -o frames/
  # 每秒保存10帧
  python video_to_frames.py input.mp4 -o frames/ --fps 10
  # 自定义文件名前缀和图片质量
  python video_to_frames.py input.mp4 -o frames/ --prefix img --quality 90
"""
    )
    parser.add_argument("video", help="输入的MP4视频文件路径")
    parser.add_argument("-o", "--output", default="frames", help="输出目录(默认: frames)")
    parser.add_argument("--fps", type=float, help="保存帧率(不指定则保存所有帧)")
    parser.add_argument("--prefix", default="frame", help="输出文件名前缀(默认: frame)")
    parser.add_argument("--quality", type=int, default=95, 
                        help="JPG图片质量1-100(默认: 95)")
    args = parser.parse_args()
    # 验证质量参数
    if args.quality < 1 or args.quality > 100:
        parser.error("图片质量必须在1-100之间")
    # 执行转换
    video_to_frames(
        video_path=args.video,
        output_dir=args.output,
        fps=args.fps,
        prefix=args.prefix,
        quality=args.quality
    )
if __name__ == "__main__":
    main()

# 计算帧间隔
if fps is not None and fps < video_fps:
    frame_interval = int(video_fps / fps)
    print(f"每隔 {frame_interval} 帧保存一次(约 {fps} FPS)")

encode_param = [int(cv2.IMWRITE_JPEG_QUALITY), quality]
cv2.imwrite(output_path, frame, encode_param)

output_path = os.path.join(output_dir, f"{prefix}_{saved_count:06d}.jpg")
# 结果：frame_000001.jpg, frame_000002.jpg ...

if saved_count % 100 == 0:
    print(f"已保存 {saved_count} 张图片...")

python video_to_frames.py my_video.mp4 -o frames/

python video_to_frames.py 4k_video.mp4 -o preview/ --fps 2 --quality 85

视频信息:
  - 帧率: 30.00 FPS
  - 总帧数: 900
  - 分辨率: 1920x1080
  - 输出目录: frames/

每隔 3 帧保存一次(约 10.0 FPS)
已保存 100 张图片...
已保存 200 张图片...
已保存 300 张图片...

完成! 共保存 300 张图片到 frames/