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()
运行之后,我们的视频效果如下: