[人工智能]2_OpenCV读写视频

读取视频使用VideoCapture类，写视频使用VideoWriter类。

读取视频时可以从视频文件读取图像，也可以从摄像头读取图像。
可以使用VideoCapture类的构造函数打开视频文件或摄像头。
如果已经创建了VideoCapture对象，可以使用VideoCapture::open()函数打开，该函数会自动调用VideoCapture::release()函数，先释放已经打开的视频，然后再打开新视频。
读取一帧使用VideoCapture::read()函数，也可以使用操作符>>。

写视频时需要设置一系列参数，包括：文件名、编解码器、帧率、宽度和高度等，编解码器使用四个字符表示，可以是CV_FOURCC('M','J','P','G')、CV_FOURCC('X,'V','I','D')、CV_FOURCC('D','I','V','X')。
写入图像时可以使用VideoWriter::write()函数，也可以使用重载符<<。
需要注意的是待写入的图像尺寸必须是与创建视频时指定的尺寸一致。

参考代码：

#include "stdafx.h"
#include <stdio.h>
#include <opencv2/highgui/highgui.hpp>

int _tmain(int argc, _TCHAR* argv[])
{
	std::string filename = "WriteVideo.avi";
	
	// 写视频
	// 设置视频的大小
	cv::Size s(320, 240);
	cv::VideoWriter writer = cv::VideoWriter(filename, CV_FOURCC('M', 'J', 'P', 'G'), 30, s);
	if (!writer.isOpened()){
		printf("创建视频文件失败！");
		return 0;
	}
	cv::Mat frame_write(s, CV_8UC3);
	for (int i = 0; i < 100; i++){
		frame_write = cv::Scalar(0);  // 背景置为黑色
		char szText[128];
		sprintf_s(szText, sizeof(szText), "%d", i);
		cv::putText(frame_write, szText, cv::Point(s.width / 3, s.height / 3), cv::FONT_HERSHEY_SCRIPT_SIMPLEX, 3, cv::Scalar(255, 0, 255), 2, 7);
		writer << frame_write;
	}

	writer.release();

	// 从视频文件读取视频
	cv::VideoCapture cap;
	cap.open(filename);
	if (!cap.isOpened()){
		printf("打开视频错误！");
		return 0;
	}
	
	cv::Mat frame;
	cv::namedWindow("video", cv::WINDOW_NORMAL);
	while (true)
	{
		cap >> frame;
		if (frame.empty())
			break;
		cv::imshow("video", frame);
		if (cv::waitKey(33) >= 0)
			break;
	}

	cv::destroyWindow("video");

	// 从摄像头读取视频
	cap.open(0);
	if (!cap.isOpened()){
		printf("打开摄像头失败！");
		return 0;
	}
	cv::Mat camera;
	cv::namedWindow("video1", cv::WINDOW_NORMAL);
	while (true){
		cap >> camera;
		if (camera.empty())
			break;
		cv::imshow("video1", camera);
		if (cv::waitKey(33) >= 0)
			break;
	}

	cv::destroyWindow("video1");

	return 0;
}