pip install opencv-python numpy matplotlib

import cv2
import os
import argparse
from datetime import timedelta
from pathlib import Path

def extract_frames(video_path, output_dir, frame_interval=None, time_interval=None, prefix="frame"):
    """
    從視頻中提取幀并保存到指定目錄

    參數(shù):
        video_path: 視頻文件路徑
        output_dir: 輸出目錄
        frame_interval: 幀間隔（每多少幀提取一幀）
        time_interval: 時間間隔（秒）
        prefix: 輸出文件名前綴
    """

    # 檢查視頻文件
    if not os.path.isfile(video_path):
        print(f"錯誤: 視頻文件 '{video_path}' 不存在")
        return

    # 創(chuàng)建輸出目錄 確保輸出目錄存在
    os.makedirs(output_dir, exist_ok=True)

    # 打開視頻文件
    cap = cv2.VideoCapture(video_path)
    if not cap.isOpened():
        print(f"錯誤: 無法打開視頻文件 {video_path}")
        return

    # 獲取視頻文件名（不含擴展名）作為默認前綴
    if prefix is None:
        video_name = Path(video_path).stem  # 例如: "input" from "input.mp4"
        prefix = video_name

    # 獲取視頻屬性
    fps = cap.get(cv2.CAP_PROP_FPS)
    frame_count = 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))
    duration = frame_count / fps

    print(f"視頻信息:")
    print(f"  視頻路徑: {video_path}")
    print(f"  幀率: {fps:.2f} fps")
    print(f"  總幀數(shù): {frame_count}")
    print(f"  分辨率: {width}x{height}")
    print(f"  時長: {timedelta(seconds=duration)}")

    # 確定提取間隔
    if frame_interval is not None:
        interval = frame_interval
        use_frame_interval = True
        print(f"將每 {interval} 幀提取一幀")
    elif time_interval is not None:
        interval = max(1, int(time_interval * fps))  # 確保至少1幀
        use_frame_interval = True
        print(f"將每 {time_interval} 秒({interval}幀)提取一幀")
    else:
        interval = 1
        use_frame_interval = True
        print("將提取所有幀")

    # 開始提取幀
    frame_number = 0
    saved_count = 0

    print("\n開始提取幀...")

    while True:
        ret, frame = cap.read()
        if not ret:
            break

        # 檢查是否應(yīng)該保存當前幀
        if use_frame_interval and frame_number % interval == 0:
            save_path = os.path.join(output_dir, f"{prefix}_{frame_number:06d}.jpg")
            cv2.imwrite(save_path, frame)
            saved_count += 1

        frame_number += 1

    cap.release()
    print(f"完成提取: 共處理 {frame_number} 幀，保存 {saved_count} 張到 {output_dir}")


def main():
    parser = argparse.ArgumentParser(description='從視頻中提取幀并保存')
    parser.add_argument('--video_dir', help='視頻所在目錄（處理目錄下所有視頻）')
    parser.add_argument('--video_path', help='單個視頻文件路徑（與video_dir互斥）')
    parser.add_argument('--output_dir', help='輸出目錄')
    group = parser.add_mutually_exclusive_group()
    group.add_argument('--frame_interval', type=int, help='幀間隔（每多少幀提取一幀）')
    group.add_argument('--time_interval', type=float, help='時間間隔（秒）')
    parser.add_argument('--prefix', help='輸出文件名前綴（默認為視頻文件名）')
    parser.add_argument('--extensions', default='mp4,mkv,avi,mov', help='允許的視頻擴展名，逗號分隔')

    args = parser.parse_args()

    # 驗證輸入?yún)?shù)
    if not args.video_dir and not args.video_path:
        parser.error("請指定 --video_dir 或 --video_path")

    # 獲取視頻文件列表
    video_files = []
    if args.video_dir:
        if not os.path.exists(args.video_dir):
            print(f"錯誤: 視頻目錄 {args.video_dir} 不存在")
            return

        allowed_extensions = set(args.extensions.lower().split(','))
        for entry in os.scandir(args.video_dir):
            if entry.is_file() and entry.name.lower().split('.')[-1] in allowed_extensions:
                video_files.append(entry.path)

        if not video_files:
            print(f"錯誤: 在目錄 {args.video_dir} 中未找到支持的視頻文件")
            return
    else:
        if not os.path.exists(args.video_path):
            print(f"錯誤: 視頻文件 {args.video_path} 不存在")
            return
        video_files = [args.video_path]

    print(f"找到 {len(video_files)} 個視頻文件需要處理")

    # 處理每個視頻
    for video_path in video_files:
        video_name = Path(video_path).stem

        # 使用視頻名作為默認前綴（除非用戶指定了全局前綴）
        prefix = args.prefix if args.prefix else video_name

        extract_frames(
            video_path,
            args.output_dir,  # 直接使用用戶指定的輸出目錄，不再創(chuàng)建子目錄
            args.frame_interval,
            args.time_interval,
            prefix
        )

if __name__ == "__main__":
    main()

python [代碼名] --video_path 視頻路徑 --output_dir 輸出目錄 [其他參數(shù)]

python frame.py --video_path input.mp4 --output_dir frames/ --time_interval 1.0

python frame.py --video_dir 視頻目錄 --output_dir 輸出目錄 [其他參數(shù)]

python frame.py --video_dir videos/ --output_dir frames/ --frame_interval 30

python frame.py --help