【Python3-OpenCV】实现图像处理—读取视频文件篇

·  阅读 474
【Python3-OpenCV】实现图像处理—读取视频文件篇

这是我参与更文挑战的第3天,活动详情查看: 更文挑战

OpenCV是一个C++库,目前流行的计算机视觉编程库,用于实时处理计算机视觉方面的问题,它涵盖了很多计算机视觉领域的模块。在Python中常使用OpenCV库实现图像处理。

image.png

本文将介绍如何在Python3中使用OpenCV实现对视频文件的读与写:

OpenCV提供了VideoCapture类和VideoWriter类来支持各种格式的视频文件的读写,在不同系统上可能支持的格式会有不同,但总的来说是都支持AVI格式的视频的。

在视频没有结束之前,可以通过VideoCapture类里面的read()方法来读取每一帧图像,每帧都是一个BGR格式的图像。

可以通过VideoWriter类里面的write()方法来把图像信息保存到VideoWriter类指向的文件中。

代码如下所示:

import cv2
videoCapture = cv2.VideoCapture("E:\\demo\\0603\\1.mp4")

#获取帧率和大小
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(
    "E:\\demo\\0603\\1.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()
复制代码

结果输出为:

image.png

结果说明:

该代码读取了原视频文件1.mp4的帧,并采用YUA颜色编码将其写入了另外一个视频帧文件1.avi

需要注意的是:

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

帧速率和帧大小也必须要指定,因为需要从另一个视频文件复制视频帧,这些属性可以通过VideoCapture类的get()函数得到。

本月将陆续推出相关系列文章,

篇篇精彩,尽请关注。

分类:
后端
标签:
收藏成功!
已添加到「」, 点击更改