在Google Colab中使用cv2处理视频后无法保存下载文件的问题

针对你遇到的代码运行无报错但找不到output.mp4的情况，可按以下步骤排查和修复：

1. 确认文件写入路径 #

在代码开头添加以下代码，查看当前工作目录并验证文件是否存在：

import os
print("当前工作目录:", os.getcwd())
# 运行后执行!ls命令查看目录内容

如果文件未出现在预期目录，可指定绝对路径写入：

out = cv2.VideoWriter('/content/output.mp4', fourcc, 30.0, (frame_width, frame_height))

2. 匹配输入与输出视频的帧尺寸 #

OpenCV的VideoWriter要求输出帧尺寸必须与输入视频完全一致，否则会静默写入失败。替换输出视频初始化代码为：

# 获取输入视频的实际帧尺寸
frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
# 用实际尺寸初始化VideoWriter
out = cv2.VideoWriter('output.mp4', fourcc, 30.0, (frame_width, frame_height))

3. 修复循环中的帧读取逻辑 #

原代码会在视频读取结束后（ret=False）仍执行一次out.write()，写入无效帧可能导致文件损坏或无法生成。调整循环为：

cnt = 0
while True:
    ret, frame = cap.read()
    # 读取不到帧时退出循环
    if not ret:
        break
    print(cnt, end=' ')

    # 原有的帧处理代码
    match = np.where(prey_frames==cnt)[0]
    if match:
        prey_frame = match[0]
        image = cv2.circle(frame,(int(prey_px[prey_frame].x),95+int(prey_px[prey_frame].y)),
                           radius=5,color=(255,0,255),thickness=2)
    else:
        image = frame

    out.write(image)
    cnt += 1

4. 更换视频编码格式 #

Colab环境中H264编码可能存在兼容性问题，换成MP4V编码尝试：

fourcc = cv2.VideoWriter_fourcc(*'MP4V')

5. 确保资源正确释放 #

用try-finally块保证即使循环中出现异常，视频写入器和捕获器也能正确释放：

cap = cv2.VideoCapture(vid_file)
fourcc = cv2.VideoWriter_fourcc(*'MP4V')
frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
out = cv2.VideoWriter('output.mp4', fourcc, 30.0, (frame_width, frame_height))

cnt = 0
try:
    while True:
        ret, frame = cap.read()
        if not ret:
            break
        print(cnt, end=' ')

        match = np.where(prey_frames==cnt)[0]
        if match:
            prey_frame = match[0]
            image = cv2.circle(frame,(int(prey_px[prey_frame].x),95+int(prey_px[prey_frame].y)),
                               radius=5,color=(255,0,255),thickness=2)
        else:
            image = frame

        out.write(image)
        cnt += 1
finally:
    out.release()
    cap.release()
    cv2.destroyAllWindows()

完成以上修改后，重新运行代码，再用!ls命令查看当前目录，即可找到output.mp4，此时使用files.download('output.mp4')即可下载文件。

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