1. 程式人生 > >OpenCV_讀取資料夾下的圖片生成視訊檔案

OpenCV_讀取資料夾下的圖片生成視訊檔案

使用OpenCV讀取指定資料夾下的所有圖片,然後生成avi視訊檔案。

//  讀取資料夾下的圖片生成視訊檔案
//  Author:www.icvpr.com
//  Blog:  http://blog.csdn.net/icvpr  

#include <iostream>
#include <string>
#include <io.h>

#include <opencv2/opencv.hpp>

using namespace std;
using namespace cv;

int main(int argc, char** argv)
{
	// 圖片集
	string fileFolderPath = "..\\images";
	string fileExtension = "jpg";
	string fileFolder = fileFolderPath + "\\*." + fileExtension;

	// 輸出視訊
	string outputVideoName = "output.avi";
	
	// openCV video writer
	VideoWriter writer;
	
	int codec = 0;
	int frameRate = 25;
	Size frameSize;


	// 遍歷資料夾
	char fileName[1000];

	struct _finddata_t fileInfo;    // 檔案資訊結構體

	// 1. 第一次查詢
	long findResult = _findfirst(fileFolder.c_str(), &fileInfo);          
	if (findResult == -1)
	{
		_findclose(findResult); 
		return -1;
	}
	
	// 2. 迴圈查詢
	do 
	{
		sprintf(fileName, "%s\\%s", fileFolderPath.c_str(), fileInfo.name);

		if ( fileInfo.attrib == _A_ARCH)  // 是存檔型別檔案
		{		
			Mat frame;
			frame = imread(fileName);    // 讀入圖片
			if (!writer.isOpened())
			{
				frameSize.width  = frame.cols;
				frameSize.height = frame.rows;
	
				if (!writer.open(outputVideoName, CV_FOURCC('D','I','V','X') , frameRate, frameSize, true))
				{
					cout << "open writer error..." << endl;
					return -1;
				}
			}

			// 將圖片資料寫入
			writer.write(frame);

			// 顯示
			imshow("video", frame);
			waitKey(frameRate);
		}

	} while (!_findnext(findResult, &fileInfo));  


	_findclose(findResult); 


	return 0;
}


如果生成的avi視訊的檔案大小為0KB的話, 說明視訊生成失敗,可能的原因是編碼器沒有。下載安裝Xvid編解碼器後,再執行試一下,一般沒有問題。

下載地址: http://www.xvidmovies.com/codec/