import cv2
import os

def video_to_images_basic(
    video_path,          # 輸入視頻文件路徑（如：test.mp4）
    output_dir,          # 輸出圖片文件夾路徑
    frame_interval=30,   # 提取間隔：每30幀提取1張（默認30幀，對應1秒/幀，可調整）
    img_format="jpg",    # 輸出圖片格式（jpg/png，推薦jpg，占用空間?。?
    img_prefix="frame"   # 圖片文件名前綴（如：frame_0001.jpg）
):
    # 1. 創(chuàng)建輸出文件夾（若不存在則自動創(chuàng)建）
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)
        print(f"創(chuàng)建輸出文件夾：{output_dir}")

    # 2. 打開視頻文件
    cap = cv2.VideoCapture(video_path)
    if not cap.isOpened():
        raise ValueError(f"無法打開視頻文件：{video_path}")

    # 3. 獲取視頻基本信息
    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))  # 視頻總幀數(shù)
    fps = cap.get(cv2.CAP_PROP_FPS)                        # 視頻幀率（每秒幀數(shù)）
    print(f"視頻信息：總幀數(shù)={total_frames}，幀率={fps:.2f}，預計提取圖片數(shù)={total_frames//frame_interval + 1}")

    # 4. 循環(huán)提取幀并保存
    frame_count = 0  # 當前幀計數(shù)器
    saved_count = 0  # 已保存圖片計數(shù)器

    while True:
        # 讀取一幀視頻
        ret, frame = cap.read()

        # 若讀取失?。ㄒ训揭曨l末尾），退出循環(huán)
        if not ret:
            break

        # 按間隔提取幀
        if frame_count % frame_interval == 0:
            # 構造圖片文件名（補零對齊，方便排序）
            img_name = f"{img_prefix}_{saved_count:04d}.{img_format}"
            img_path = os.path.join(output_dir, img_name)

            # 保存圖片
            cv2.imwrite(img_path, frame)
            print(f"已保存：{img_path}")

            saved_count += 1

        frame_count += 1

    # 5. 釋放資源
    cap.release()
    print(f"\n轉換完成！共保存 {saved_count} 張圖片，存儲路徑：{output_dir}")

# ------------------- 調用示例 -------------------
if __name__ == "__main__":
    # 配置參數(shù)
    VIDEO_PATH = "input_video.mp4"  # 你的視頻文件路徑
    OUTPUT_DIR = "extracted_images" # 圖片輸出文件夾
    FRAME_INTERVAL = 30             # 每30幀提取1張（1秒1張，若視頻幀率25，則每1.2秒1張）

    # 執(zhí)行轉換
    video_to_images_basic(
        video_path=VIDEO_PATH,
        output_dir=OUTPUT_DIR,
        frame_interval=FRAME_INTERVAL,
        img_format="jpg"
    )

import cv2
import os

def video_to_images_by_time(
    video_path,          # 輸入視頻文件路徑
    output_dir,          # 輸出圖片文件夾路徑
    time_interval=1.0,   # 時間間隔：每1.0秒提取1張（可調整，如2.0=每2秒1張）
    img_format="jpg",    # 輸出圖片格式
    img_prefix="frame"   # 圖片文件名前綴
):
    # 1. 創(chuàng)建輸出文件夾
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)
        print(f"創(chuàng)建輸出文件夾：{output_dir}")

    # 2. 打開視頻文件
    cap = cv2.VideoCapture(video_path)
    if not cap.isOpened():
        raise ValueError(f"無法打開視頻文件：{video_path}")

    # 3. 獲取視頻基本信息
    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
    fps = cap.get(cv2.CAP_PROP_FPS)
    video_duration = total_frames / fps  # 視頻總時長（秒）
    frame_interval = int(fps * time_interval)  # 換算為幀間隔

    print(f"視頻信息：總幀數(shù)={total_frames}，幀率={fps:.2f}，總時長={video_duration:.2f}秒")
    print(f"時間間隔={time_interval}秒，對應幀間隔={frame_interval}幀，預計提取圖片數(shù)={int(video_duration//time_interval) + 1}")

    # 4. 循環(huán)提取幀并保存
    frame_count = 0
    saved_count = 0
    last_saved_time = 0.0  # 上一次保存圖片的時間

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

        # 獲取當前幀對應的時間（秒）
        current_time = frame_count / fps

        # 按時間間隔保存
        if current_time - last_saved_time >= time_interval:
            img_name = f"{img_prefix}_{saved_count:04d}.{img_format}"
            img_path = os.path.join(output_dir, img_name)
            cv2.imwrite(img_path, frame)
            print(f"時間{current_time:.2f}秒：已保存 {img_path}")

            saved_count += 1
            last_saved_time = current_time

        frame_count += 1

    # 5. 釋放資源
    cap.release()
    print(f"\n轉換完成！共保存 {saved_count} 張圖片，存儲路徑：{output_dir}")

# ------------------- 調用示例 -------------------
if __name__ == "__main__":
    VIDEO_PATH = "input_video.mp4"
    OUTPUT_DIR = "extracted_images_by_time"
    TIME_INTERVAL = 2.0  # 每2秒提取1張圖片

    video_to_images_by_time(
        video_path=VIDEO_PATH,
        output_dir=OUTPUT_DIR,
        time_interval=TIME_INTERVAL,
        img_format="jpg"
    )