OpenCV提供了VideoCapture类和VideoWriter类来支持各种格式的视频文件的读写,在不同系统上可能支持的格式会有不同,但总的来说是都支持AVI格式的视频的。
在视频没有结束之前,可以通过VideoCapture类里面的read()方法来读取每一帧图像,每帧都是一个BGR格式的图像。
可以通过VideoWriter类里面的write()方法来把图像信息保存到VideoWriter类指向的文件中。
如下示例:
import cv2
videoCapture = cv2.VideoCapture("F:\\Python\\picture\\Megamind.avi")
#获取帧率和大小
fps = videoCapture.get(cv2.CAP_PROP_FPS)
size = (int(videoCapture.get(cv2.CAP_PROP_FRAME_WIDTH)),
int(videoCapture.get(cv2.CAP_PROP_FRAME_HEIGHT)))
#设置输出的视频信息(视频文件名,编解码器,帧率,大小)
videoWriter = cv2.VideoWriter(
"myTestVideo.avi",cv2.VideoWriter_fourcc('I','4','2','0'),fps,size)
#读取视频文件,如果要读取的视频还没有结束,那么success接收到的就是True,每一帧的图片信息保存在frame中,通过write方法写到指定文件中
success,frame = videoCapture.read()
while success:
videoWriter.write(frame)
success, frame = videoCapture.read()
需要注意的是:VideoWriter类的构造方法必须指定视频的文件名,这个文件名如果存在就覆盖它。也必须指定视频编解码器。编解码器的可行性根据系统不同而不同。需要指定帧率和大小,这个可以通过get()方法来获得。
常见的编解码器:
- cv2.VideoWriter_fourcc(‘I’,‘4’,‘2’,‘0’)。该选项是一个未压缩的YUV颜色编码,是4:2:0的色度子采样,这种编码方式有很好的兼容性,但会产生比较大的文件,文件扩展名为 .avi。
- cv2.VideoWriter_fourcc(‘P’,‘I’,‘M’,‘1’)。该选项是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’,‘1’)。该选项是一个Flash视频,文件扩展名应为 .flv。