OpenCV视频写入空文件：尺寸参数的常见陷阱与解决方案

cv2.VideoWriter(filename, fourcc, fps, frameSize[, isColor])

import cv2
import os # 用于检查文件是否存在，非必需

def create_video_from_images(image_prefix, start_index, end_index, output_filename, fps):
    """
    使用OpenCV从一系列图片生成视频文件。

    Args:
        image_prefix (str): 图片文件名前缀，例如 "capture."
        start_index (int): 图片文件名的起始索引（包含）。
        end_index (int): 图片文件名的结束索引（包含）。
        output_filename (str): 输出视频文件名，例如 "py_test.mkv" 或 "output.mp4"。
        fps (float): 视频帧率。
    """
    # 1. 读取第一张图片以获取正确的帧尺寸
    # 假设图片命名为 capture.0.jpg, capture.1.jpg, ...
    first_image_path = f"{image_prefix}{start_index}.jpg"
    first_img = cv2.imread(first_image_path)

    if first_img is None:
        print(f"错误：无法读取第一张图片 {first_image_path}。请检查路径和文件名。")
        return

    # 核心修正：frame_size 必须是 (宽度, 高度)
    # img.shape 返回 (高度, 宽度, 通道数)
    frame_height, frame_width, _ = first_img.shape
    frame_size = (frame_width, frame_height)
    print(f"检测到图片尺寸：宽度={frame_width}, 高度={frame_height}。")

    # 2. 定义视频编码器 (FOURCC)
    # 不同的编码器和容器组合有不同的兼容性。
    # 常用且兼容性较好的组合：
    # - 'XVID' 或 'DIVX' for .avi
    # - 'mp4v' 或 'H264' for .mp4
    # - 'x264' for .mkv (通常需要系统安装H.264编码器支持)
    # 注意：某些编码器可能需要系统额外安装相应的库。

    # 示例使用 'XVID'，通常兼容性较好，适用于 .avi
    # fourcc = cv2.VideoWriter.fourcc(*'XVID') 

    # 示例使用 'mp4v'，适用于 .mp4
    # fourcc = cv2.VideoWriter.fourcc(*'mp4v') 

    # 示例使用 'x264'，适用于 .mkv 或 .mp4，如果系统支持H.264编码
    # 确保输出文件扩展名与选择的fourcc兼容
    fourcc = cv2.VideoWriter.fourcc(*'x264') 

    # 3. 初始化 VideoWriter 对象
    print(f"尝试初始化 VideoWriter: 文件='{output_filename}', 编码='{fourcc}', FPS={fps}, 尺寸={frame_size}")
    out = cv2.VideoWriter(output_filename, fourcc, fps, frame_size)

    # 检查 VideoWriter 是否成功打开
    if not out.isOpened():
        print(f"错误：无法打开视频写入器 '{output_filename}'。")
        print("可能原因：")
        print("  - 指定的 FourCC 编码器不受系统支持。")
        print("  - 输出文件路径无效或没有写入权限。")
        print("  - 输出文件扩展名与 FourCC 编码器不兼容。")
        print("  - 尝试更换 FourCC 编码器或输出文件扩展名。")
        return

    # 4. 遍历图片并写入视频
    print(f"开始写入 {end_index - start_index + 1} 帧到视频...")
    for i in range(start_index, end_index + 1):
        img_path = f"{image_prefix}{i}.jpg"
        img = cv2.imread(img_path)

        if img is None:
            print(f"警告：无法读取图片 {img_path}。跳过此帧。")
            continue

        # 确保读取的图片尺寸与VideoWriter期望的尺寸一致
        # 如果不一致，可以选择跳过或进行resize
        if img.shape[1] != frame_size[0] or img.shape[0] != frame_size[1]:
            print(f"警告：图片 {img_path} 尺寸为 {img.shape[1]}x{img.shape[0]}，与视频预期尺寸 {frame_size[0]}x{frame_size[1]} 不符。将进行缩放。")
            img = cv2.resize(img, frame_size)

        out.write(img)
        # print(f"已写入帧 {i}")

    # 5. 释放 VideoWriter 对象
    # 这一步至关重要，它会关闭文件并确保所有缓存的帧都已写入。
    out.release()
    print(f"视频 '{output_filename}' 已成功创建。")

# --- 示例用法 ---
if __name__ == "__main__":
    # 请确保在运行此脚本前，在当前目录下有 'capture.0.jpg' 到 'capture.120.jpg' 这些图片文件。
    # 可以通过以下方式生成一些虚拟图片用于测试：
    # import numpy as np
    # for i in range(121):
    #     # 创建一个 120高 x 160宽 的黑色图片
    #     dummy_image = np.zeros((120, 160, 3), dtype=np.uint8)
    #     # 可选：在图片上绘制一些内容以便区分
    #     cv2.putText(dummy_image, str(i), (50, 60), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2, cv2.LINE_AA)
    #     cv2.imwrite(f"capture.{i}.jpg", dummy_image)
    # print("虚拟图片已生成。")

    image_prefix = "capture."
    start_index = 0
    end_index = 120 # 如果是 capture.0.jpg 到 capture.120.jpg，共121张
    output_video_file = "py_test.mkv" # 也可以是 "py_test.mp4", "py_test.avi"
    frame_rate = 60.0

    create_video_from_images(image_prefix, start_index, end_index, output_video_file, frame_rate)

    # 验证文件是否生成且大小正常（大于几百字节）
    if os.path.exists(output_video_file):
        file_size = os.path.getsize(output_video_file)
        print(f"生成的视频文件 '{output_video_file}' 大小为 {file_size} 字节。")
        if file_size < 1024: # 简单判断，实际视频文件应远大于此
            print("警告：文件大小异常，可能仍存在问题。")
    else:
        print(f"错误：视频文件 '{output_video_file}' 未生成。")