OpenCV | 图片与视频的相互转换（C++&Python)

Posted 2021-04-25 人工智能LeadAI

七月 上海| 高性能计算之GPU CUDA培训

7月27-29日

三天密集式学习  快速带你入门

>

正文共1338个字，预计阅读时间6分钟。

前言

在学习opencv的过程中，终会遇到需要把视频转换成图片处理，或者把处理后的图片保存为视频格式的时候。这篇文章中就来看一下如何进行图片和视频之间的相互转换。

视频转图片

对于这种情况大家应该很熟悉了。因为学习opencv的时候肯定打开过摄像头，比如打开摄像头检测个人脸或者笑脸之类的。在这个过程中，实际就是把摄像头拍摄到的视频转换为一帧一帧的图片连续地进行处理。

我们知道，对于打开视频和摄像头，opencv提供了VideoCapture类，可以方便地通过流操作符>>与Mat类进行交互。这种情况比较常见，我们直接看例子。

 1#include<opencv2\opencv.hpp>
 2using namespace cv;
 3int main()
 4{
 5//打开默认摄像头
 6VideoCapture cap(0); 
 7if (!cap.isOpened())
 8{
 9return -1;
10}
11Mat frame;
12// 按Q键退出时，键盘需要调为英文模式
13while(waitKey(30) != 'q')
14{
15// 通过流操作符把视频转化为一帧帧图片
16cap >> frame;
17// Do something here !
18imshow("video", frame);
19}
20return 0;
21}

如果是要打开视频文件而不是摄像头，只需要像下面这样：

1VideoCapture cap("E:\\我的视频\\视频.mp4");

而视频转成图片也就是那么简单的一句cap >> frame;

图片转视频

视频转图片只需要流操作符简单的一句代码就能实现，那么图片转视频呢？是不是也是这么简单？

答案是肯定的。也是只需要用一句这么简单的代码，而且只需要把流操作符的方向反过来就行了。这时候用到的就是opencv提供的VideoWriter类。VideoCapture和VideoWriter，顾名思义，他俩一个是抓取视频的，一个是写入视频的。

VideoWriter构造函数如下所示：

VideoWriter(const string& filename, int fourcc, double fps, Size frameSize, bool isColor=true)

各参数含义如下：

• filename： 输出视频文件的名称

• fourcc：4-character code of codec的缩写，四个字符用来表示压缩帧的编解码。

• fps：帧率，我猜是frame per second的缩写。

• frameSize：每帧图片的大小

• isColor： 如果非零，编码器将希望得到彩色帧并进行编码；否则，是灰度帧（只有在Windows下支持这个标志）

第二个参数，OpenCV提供的格式是未经过压缩的，目前支持的格式如下：

CV_FOURCC('P', 'I', 'M', '1') = MPEG-1 codec

CV_FOURCC('M', 'J', 'P', 'G') = motion-jpeg codec

CV_FOURCC('M', 'P', '4', '2') = MPEG-4.2 codec

CV_FOURCC('D', 'I', 'V', '3') = MPEG-4.3 codec

CV_FOURCC('D', 'I', 'V', 'X') = MPEG-4 codec

CV_FOURCC('U', '2', '6', '3') = H263 codec

CV_FOURCC('I', '2', '6', '3') = H263I codec

CV_FOURCC('F', 'L', 'V', '1') = FLV1 codec

还是用一个例子来看：

 1#include<opencv2\opencv.hpp>
 2#include<iostream>
 3using namespace std;
 4using namespace cv;
 5int main()
 6{
 7// 构造一个VideoWriter
 8VideoWriter video("test.avi", CV_FOURCC('M', 'J', 'P', 'G'), 25.0, Size(640, 480));
 9// 从一个文件夹下读取多张jpg图片
10String pattern = "G:\\temp_picture\\*.jpg";
11vector<String> fn;
12glob(pattern, fn, false);
13size_t count = fn.size();
14for (size_t i = 0; i < count; i++)
15{
16Mat image = imread(fn[i]);
17// 这个大小与VideoWriter构造函数中的大小一致。
18resize(image, image, Size(640, 480));
19// 流操作符，把图片传入视频
20video << image;
21}
22cout << "处理完毕！" << endl;
23// 处理完之后会在得到一个名为test.avi的视频文件。
24}

Python代码

最后还是给出Python版本的示例代码：

视频转图片

1import cv2
2import numpy 
3cap = cv2.VideoCapture(0)
4while cv2.waitKey(30)!=ord('q'):
5retval, image = cap.read()
6cv2.imshow("video",image)
7cap.release()

图片转视频

1# encoding: UTF-8
2import glob as gb
3import cv2
4img_path = gb.glob("G:\\temp_picture\\*.jpg") 
5videoWriter = cv2.VideoWriter('test.mp4',cv2.VideoWriter_fourcc(*'MJPG'), 25, (640,480))
6for path in img_path:
7img  = cv2.imread(path) 
8img = cv2.resize(img,(640,480))
9videoWriter.write(img)

额，虽然Python代码没加注释，但是好像还是理解了什么叫人生苦短，我用Python。

原文链接：https://www.jianshu.com/p/65aa6516baa6

查阅更为简洁方便的分类文章以及最新的课程、产品信息，请移步至全新呈现的“LeadAI学院官网”：

www.leadai.org

大家都在看