视频处理VideoWriter类---OpenCV-Python开发指南（39）

VideoWriter

在前一篇博文中，我们学习了如何使用摄像头，并处理每一帧的图像。但是，处理完视频之后，并没有讲解如何保存视频。所以，本篇将详细的讲解如何保存视频中的图片以及视频本身。

在OpenCV中，它给我们提供了cv2.VideoWriter类将图片序列保存为视频。同时，通过该类，我们也可以修改视频的各种属性，完成对视频类型的转换。

一般我们使用cv2.VideoWriter类，需要如下3个步骤。

构造函数

OpenCV为cv2.VideoWriter类提供了构造函数，用它来实现初始化工作。该函数的完整定义如下：

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

filename：需要输出保存的视频文件名。如果文件名存在，覆盖原文件。

fourcc：视频的编码类型。在OpenCV中，cv2.VideoWriter_fourcc()函数用来指定视频的编码格式。该函数的参数有4个，这4个字符构成了编/解码器的“4字标记”，每个编/解码器都有一个这样的标记。常用标记如下表：

取值	含义
cv2.VideoWriter_fourcc('I','4','2','0')	未压缩的YUV颜色编码格式，色度子采样为4：2：0，该编码格式具有较好的兼容性，但生成的视频文件较大，文件扩展名为.avi
cv2.VideoWriter_fourcc('P','I','M','I')	MPEG-1编码格式，生成的文件扩展名为.avi
cv2.VideoWriter_fourcc('X','V','I','D')	MPEG-4编码格式，如果希望得到的视频大小为平均值，可以选用这个参数组合。文件扩展名为.avi
cv2.VideoWriter_fourcc('T','H','E','O')	Ogg Vorbis编码格式，文件扩展名为.ogv
cv2.VideoWriter_fourcc('F','L','V','I')	Flash视频格式，文件扩展名为.flv
cv2.VideoWriter_fourcc('M','P','4','V')	文件扩展名.mp4

fps：帧速率，比如录制视频每秒30帧，或者60帧等。

frameSize：帧的长宽

isColor：是否为彩色图像

write函数

cv2.VideoWriter类中还提供了cv2.VideoWriter.write()函数用于写入下一帧视频。其完整定义如下：

cv2.VideoWriter.write(image)

image参数是需要写入的视频帧，也就是前篇博文最后实战项目中的frame。

释放

同样的，在不需要cv2.VideoWriter类对象时，需要将其释放。释放所使用的函数为cv2.VideoWriter.release()。

实战保存摄像头录制的视频

通过上面的讲解，我们已经知道了如何保存摄像头的视频。下面，我们来实战保存摄像头录制的视频影像，具体代码如下所示：

import cv2

cap = cv2.VideoCapture(0)
fourcc = cv2.VideoWriter.fourcc('m', 'p', '4', 'v')
out = cv2.VideoWriter('output.mp4', fourcc, 20, (640, 480))
while (cap.isOpened()):
    ret, frame = cap.read()
    if ret:
        out.write(frame)
        cv2.imshow('video', frame)
        c = cv2.waitKey(1)
        if c == 27:
            break
    else:
        break
cap.release()
out.release()
cv2.destroyAllWindows()

运行之后，我们就会捕获摄像头录制的视频，直到你按下ESC键后停止。

录制Canny边缘检测视频

通过上文，我们知道如果如果后期处理视频，就需要对其每一帧的图像进行处理。现在，我们来实现Canny边缘检测，具体代码如下：

import cv2

cap = cv2.VideoCapture(0)
fourcc = cv2.VideoWriter.fourcc('m', 'p', '4', 'v')
out = cv2.VideoWriter('output.mp4', fourcc, 20, (640, 480))
while (cap.isOpened()):
    ret, frame = cap.read()
    if ret:
        frame=cv2.Canny(frame,100,200)
        out.write(frame)
        cv2.imshow('video', frame)
        c = cv2.waitKey(1)
        if c == 27:
            break
    else:
        break
cap.release()
out.release()
cv2.destroyAllWindows()

运行之后，我们的视频效果如下：

1.png