【C++】【OpenCV-4.9.0】视频写入（VideoWriter，借助samples中的代码示例来进行学习）
借助官方离线文档中的samples来理解VideoWriter
文档位置：samples/cpp/tutorial_code/videoio/video-write/video-write.cpp
注：需要提前下载 openh264-1.8.0-win64.dll，然后放在Release文件夹下 ，否则无法正确对输出文件进行编码从而运行失败
1 #include <iostream> 2 #include <string> 3 4 #include <opencv2/core.hpp> 5 #include <opencv2/videoio.hpp> 6 7 using namespace std; 8 using namespace cv; 9 10 void help() { 11 cout 12 << "------------------------------------------------------------------------------" << endl 13 << "This program shows how to write video files." << endl 14 << "You can extract the R or G or B color channel of the input video." << endl 15 << "Usage:" << endl 16 << "./video-write <input_video_name> [ R | G | B] [Y | N]" << endl 17 << "------------------------------------------------------------------------------" << endl 18 << endl; 19 } 20 21 int main(int argc, char* argv[]) { 22 help(); 23 24 // 如果命令行接收的参数长度不等于4则异常退出。 25 if (argc != 4) { 26 cout << "Not enough parameters" << endl; 27 return -1; 28 } 29 30 // 输入的视频文件，如：CV.exe AVI.avi R G ， source = "AVI.avi" 31 const string source = argv[1]; 32 // 判断用户是否询问输出类型，如果是N，将使用输入视频的编码器类型 33 const bool askOutputType = argv[3][0] == 'Y'; 34 35 VideoCapture inputVideo(source); 36 if (!inputVideo.isOpened()) { 37 cout << "Could not open the input video: " << source << endl; 38 return -1; 39 } 40 41 // 查找最后一个扩展点，返回下标索引位置，如：CV.exe AVI.avi R G ，pAt = 3 42 string::size_type pAt = source.find_last_of('.'); 43 44 // 定义输出文件名，NAME = "AVIR.avi" 45 const string NAME = source.substr(0, pAt) + argv[2][0] + ".avi"; 46 47 // 获取视频流中的第一帧的编码类型，并转换为int类型（视频流中，每一帧的编码格式可能都不一样） 48 // 例： 49 // 转换前 -- 8.75967e+08 50 // 转换后 -- 875967080 51 unsigned int ex = static_cast<int>(inputVideo.get(CAP_PROP_FOURCC)); 52 53 // 通过按位运算符从int转换为char 54 // 转换前 -- 875967080 对应的二进制：00110100 00110110 00110010 01101000 55 // 转换后 -- h264 56 char EXT[] = { 57 (char)(ex & 0XFF), 58 (char)((ex & 0XFF00) >> 8), 59 (char)((ex & 0XFF0000) >> 16), 60 (char)((ex & 0XFF000000) >> 24), 61 0 62 }; 63 64 // 获取输入视频的尺寸 65 Size S( 66 (int)inputVideo.get(CAP_PROP_FRAME_WIDTH), 67 (int)inputVideo.get(CAP_PROP_FRAME_HEIGHT) 68 ); 69 70 // 创建一个写入对象，并指定相关参数（输出视频文件名、编码格式、输入视频帧率、输出视频尺寸、是否为彩色或者灰色） 71 VideoWriter outputVideo; 72 if (askOutputType) 73 outputVideo.open(NAME, ex = -1, inputVideo.get(CAP_PROP_FPS), S, true); // 由CV自动选择帧的编码格式 74 else 75 outputVideo.open(NAME, ex, inputVideo.get(CAP_PROP_FPS), S, true); // 和输入视频中的每一帧编码格式保持一致 76 77 // 如果打开失败则输出错误并异常退出 78 if (!outputVideo.isOpened()) { 79 cout << "Could not open the output video for write: " << source << endl; 80 return -1; 81 } 82 83 // 输出视频尺寸、总帧数和输入视频的编码格式 84 cout << "Input frame resolution: Width = " << S.width << " Height = " << S.height 85 << "of nr#: " << inputVideo.get(CAP_PROP_FRAME_COUNT) << endl; 86 cout << "Input codec type: " << EXT << endl; 87 88 // 选择颜色通道 89 int channel = 2; 90 switch (argv[2][0]) { 91 case 'R':channel = 2; break; 92 case 'G':channel = 1; break; 93 case 'B':channel = 0; break; 94 } 95 96 Mat src, res; 97 vector<Mat> spl; 98 99 // 循环写入视频 100 for (;;) { 101 inputVideo >> src; 102 if (src.empty()) break; 103 104 // 分离通道 105 split(src, spl); 106 107 // 将不符合选择的颜色通道都置零处理 108 for (int i = 0; i < 3; ++i) 109 if (i != channel) 110 spl[i] = Mat::zeros(S, spl[0].type()); 111 112 // 合并通道 113 merge(spl, res); 114 115 outputVideo << res; 116 } 117 118 cout << "Finished writing" << endl; 119 return 0; 120 }
运行【CV.exe Video.avi R N】结果：
运行【CV.exe Video.avi R Y】结果：对于给出的报错说文件名有误实在没有头绪，就懒得继续深究了。
上述的注释基本解释了每行代码的作用，现在总结下运行逻辑：
上述代码功能——通过选择不同的颜色通道，将输入视频另存为选择好的单一颜色通道的视频文件。
1、判断命令行启动参数个数是否正确；
2、获取输入视频文件的首帧编码格式、视频文件尺寸；
3、通过输入参数中的【R | G | B】的选择定义输出视频文件名；
4、创建输出类对象，并通过输入参数中的最后一个来确定编码类型；
5、将输入文件根据选择好的颜色通道进行分离置零再合并，然后循环写入输出文件；
注：split对帧进行分离后，得到的三通道顺序是BGR，这就是为什么 ‘R’ 对应的是 channel[2]。
