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()

博主个人公众号
版权声明 ▶ 本网站名称:陶小桃Blog
▶ 本文链接:https://www.52txr.cn/2022/OpenCV58.html
▶ 本网站的文章部分内容可能来源于网络,仅供大家学习与参考,如有侵权,请联系站长进行核实删除。
▶ 转载本站文章需要遵守:商业转载请联系站长,非商业转载请注明出处!!

最后修改:2022 年 05 月 26 日
如果觉得我的文章对你有用,请随意赞赏