OpenCV For Python入门58 视频处理之VideoWriter类

cv2.VideoWriter类可以将图片序列保存成视频文件，也可以修改视频的各个属性，还可以完成对视频类型的转换。

类函数介绍

cv2.VideoWriter类常用的成员函数包括：构造函数、write函数等。

1、构造函数

语法格式：

<VideoWriter object> = cv2.VideoWriter( filename, fourcc, fps, frameSize, [isColor] )

• filename：指定输出目标视频的存放路径和文件名。如果已经存在，则会被覆盖

• fourcc：表示视频编/解码类型。

  • 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
• fps：为帧速率
• frameSize：为每一帧的长和宽
• isColor：表示是否为彩色

例如，下面的语句完成了cv2.VideoWriter类的初始化工作。

fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('output.avi',fourcc,20,(1024,768))

2、write函数

用于写入下一帧视频。语法格式：

None = cv2.VideoWriter.write(image)

image是要写入的视频帧。要求彩色图像的格式为BGR模式。

fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('output.avi',fourcc,20,(1024,768))
out.write(frame) #有一个视频帧为frame

3、释放

在不需要cv2.VideoWriter类对象时，需要将其释放。语法:

None = cv2.VideoWriter.release()

保存视频

保存视频包括创建对象、写入视频、释放对象等多个步骤。

1、创建对象

在创建对象前，首先需要设置好参数。

• 设置好需要保存的具体文件名，例如filename='out.avi'
• 使用cv2.VideoWriter_fourcc()确定编/解码的类型，例如：fourcc = cv2.VideoWriter_fourcc(*'XVID')
• 确定视频的帧速率，例如fps=20
• 确定视频的长度和宽度，例如：size=(640, 480)

然后利用上面设置好的参数：

out = cv2.VideoWriter(filename, fourcc, fps, size)

当然，也可以在函数内直接给参数赋值。

2、写入视频

out = cv2.VideoWriter(filename, fourcc, fps, size)
out.write(frame)

3、释放对象

out.release()

例1：使用cv2.VideoWriter类来保存摄像头视频文件。

import numpy as np
import cv2
cap = cv2.VideoCapture(0)
fourcc = cv2.VideoWriter_fourcc('I','4','2','0')
out = cv2.VideoWriter('output.avi',fourcc,20,(640,480))
while(cap.isOpened()):
    ret, frame = cap.read()
    if ret == True:
        out.write(frame)
        cv2.imshow('frame',frame)
        if cv2.waitKey == 27:
            break
    else:
        break
cap.release()
out.release()
cv2.destroyAllWindows()