解决Python OpenCV无法写入MP4视频文件的常见问题

import cv2
import numpy as np
import os

# 视频参数示例
video_output_path = "output_video.mp4"
width, height = 640, 480
fps = 30
duration_seconds = 5 # 视频时长

# 模拟一些帧数据
# 在实际应用中，这些帧将来自你的摄像头、视频文件或图像处理流程
frames = []
for i in range(int(fps * duration_seconds)):
    frame = np.zeros((height, width, 3), dtype=np.uint8) # 创建一个黑色帧
    # 在帧上添加一些文本以区分帧
    cv2.putText(frame, f"Frame: {i+1}", (50, 50), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
    cv2.putText(frame, f"Time: {i/fps:.2f}s", (50, 100), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 255), 2)
    frames.append(frame)

print(f"准备写入视频到: {video_output_path}")
print(f"视频尺寸: {width}x{height}, 帧率: {fps} FPS")

# 尝试不同的FourCC编码器
# 1. 尝试 'XVID' (推荐用于MP4或AVI，兼容性较好)
fourcc_codes_to_try = ['XVID', 'H264', 'mp4v'] # 尝试顺序
success = False

for fourcc_str in fourcc_codes_to_try:
    print(f"\n尝试使用 '{fourcc_str}' 编码器...")
    try:
        # FourCC代码通常需要转换为整数形式
        fourcc = cv2.VideoWriter_fourcc(*fourcc_str)

        # 初始化VideoWriter
        video_writer = cv2.VideoWriter(video_output_path, fourcc, fps, (width, height))

        # 检查VideoWriter是否成功打开
        if not video_writer.isOpened():
            print(f"错误：无法使用 '{fourcc_str}' 编码器打开视频写入器。")
            print("请检查文件路径、写入权限，以及FFmpeg是否正确安装并支持此编码器。")
            continue # 尝试下一个编码器

        # 写入所有帧
        for frame in frames:
            video_writer.write(frame)

        # 释放VideoWriter对象，确保所有缓冲数据写入文件
        video_writer.release()

        # 检查文件是否生成且大小不为0
        if os.path.exists(video_output_path) and os.path.getsize(video_output_path) > 0:
            print(f"视频已成功写入：{video_output_path} (使用 '{fourcc_str}' 编码器)")
            success = True
            break # 成功写入，退出循环
        else:
            print(f"警告：使用 '{fourcc_str}' 编码器写入成功，但文件 {video_output_path} 大小为0KB或未生成。")
            print("这通常意味着编码器不兼容或FFmpeg配置有问题。")

    except Exception as e:
        print(f"使用 '{fourcc_str}' 编码器写入视频时发生错误：{e}")

if not success:
    print("\n所有尝试的编码器均未能成功写入有效的MP4文件。")
    print("请确保FFmpeg已正确安装并配置在系统PATH中，并考虑更新OpenCV版本。")

# 示例：原始问题中能正常工作的AVI写入方式
avi_output_path = "output_avi.avi"
print(f"\n原始工作示例：尝试使用 'MJPG' 编码器写入 AVI 到 {avi_output_path}...")
try:
    fourcc_mjpg = cv2.VideoWriter_fourcc(*'MJPG')
    video_writer_avi = cv2.VideoWriter(avi_output_path, fourcc_mjpg, fps, (width, height))

    if not video_writer_avi.isOpened():
        print(f"错误：无法使用 'MJPG' 编码器打开AVI写入器。")
    else:
        for frame in frames:
            video_writer_avi.write(frame)
        video_writer_avi.release()
        if os.path.exists(avi_output_path) and os.path.getsize(avi_output_path) > 0:
            print(f"AVI视频已成功写入：{avi_output_path} (使用 'MJPG' 编码器)")
        else:
            print(f"警告：使用 'MJPG' 编码器写入AVI成功，但文件 {