在Google Colab中用OpenCV生成的视频损坏问题求助

你的代码执行无报错但生成无效视频，核心原因大概率是VideoWriter初始化失败、帧色彩空间不匹配或编码格式不兼容，以下是具体解决步骤：

1. 先检查VideoWriter是否正常初始化 #

OpenCV的VideoWriter初始化失败时不会抛出异常，但后续的write()操作完全无效，最终生成极小的空文件。必须添加初始化校验：

height, width, layers = cropped_faces[0].shape
print(height, width, layers)

fourcc = cv2.VideoWriter_fourcc(*'H264')
video = cv2.VideoWriter('/content/gdrive/MyDrive/Kaggle/Data/output.mp4', fourcc, 1, (width, height))

# 新增：检查写入器是否成功打开
if not video.isOpened():
    print("视频写入器初始化失败！请检查路径是否存在、编码格式是否支持或尺寸参数是否正确")
    exit()

如果这里打印错误，优先排查路径权限（比如Google Drive挂载是否正常）或编码格式问题。

2. 转换帧的色彩空间：RGB→BGR #

Deepface输出的人脸图像是RGB格式，但OpenCV的VideoWriter要求输入帧为BGR格式。直接写入RGB帧会导致编码异常，生成无法播放的视频。修改写入逻辑：

for j in range(0,5):
    # 将RGB帧转成OpenCV需要的BGR格式
    bgr_frame = cv2.cvtColor(cropped_faces[j], cv2.COLOR_RGB2BGR)
    video.write(bgr_frame)

3. 更换兼容的编码格式 #

H264编码在Colab等环境中常因依赖库缺失无法正常工作，换成更通用的编码：

方案1：用MP4V编码（适配MP4格式） #

fourcc = cv2.VideoWriter_fourcc(*'MP4V')

方案2：用XVID编码（生成AVI格式） #

如果MP4格式仍有问题，试试生成AVI文件：

fourcc = cv2.VideoWriter_fourcc(*'XVID')
video = cv2.VideoWriter('/content/gdrive/MyDrive/Kaggle/Data/output.avi', fourcc, 1, (width, height))

4. 确保所有帧尺寸一致 #

如果cropped_faces里的帧宽高不一致，VideoWriter会静默丢弃不匹配的帧，最终视频内容缺失。可以提前校验并统一尺寸：

# 遍历所有帧检查并修正尺寸
for idx, face in enumerate(cropped_faces):
    h, w, _ = face.shape
    if h != height or w != width:
        print(f"第{idx}帧尺寸不匹配，已自动调整为{width}x{height}")
        cropped_faces[idx] = cv2.resize(face, (width, height))

完整修正代码 #

import cv2

# 假设cropped_faces已通过Deepface正确获取
height, width, layers = cropped_faces[0].shape
print(f"帧尺寸：{width}x{height}，通道数：{layers}")

# 使用兼容的MP4V编码
fourcc = cv2.VideoWriter_fourcc(*'MP4V')
video_path = '/content/gdrive/MyDrive/Kaggle/Data/output.mp4'
video = cv2.VideoWriter(video_path, fourcc, 1, (width, height))

# 校验写入器初始化状态
if not video.isOpened():
    print(f"无法创建视频文件：{video_path}，请检查路径权限或编码格式")
    exit()

# 统一帧尺寸并转换色彩空间后写入
for idx in range(0,5):
    face = cropped_faces[idx]
    # 修正尺寸
    h, w, _ = face.shape
    if h != height or w != width:
        face = cv2.resize(face, (width, height))
    # RGB转BGR
    bgr_face = cv2.cvtColor(face, cv2.COLOR_RGB2BGR)
    video.write(bgr_face)

video.release()
cv2.destroyAllWindows()

print("视频已成功保存")

内容的提问来源于stack exchange，提问作者Pawara Siriwardhane