视频处理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

你可能感兴趣的:(视频处理VideoWriter类---OpenCV-Python开发指南(39))