OpenCV 视频与图片序列相互转换
我们在计算机视觉相关的实验中都需要视频与图片序列之间相互转换，在平时的实验中，经常需要将视频保存为一帧一帧的图片，以获取实验结果；另一方面，很多标准的算法测试数据库都是图片序列（文件名以帧号+图片扩展名），我们可能需要将其转换为视频，因此视频与图片之间的转换是很常见的操作。在网上找到比较好的测试程序，一方面将其封装为函数，方便调用；另一方面，视频转图片可以任意设定图片输出目录，函数会自动创建相关目录；图片序列转视频可以设定更多的参数，也就可以更好的控制，而且可以自行设定视频输出的目录，函数自动检测并创建相关目录。适合windows平台+OpenCV；
代码如下：
View Code
1 // ImgToVideo_test.cpp : Defines the entry point for the console application. 2 // 3 4 #include "stdAfx.h" 5 #include <cv.h> 6 #include <highgui.h> 7 #include <iostream> 8 #include <direct.h>//for mk_dir 9 #include <io.h>//for _acess() 10 #include <string> 11 12 int recursive_mkdir( char *dir );//创建多级目录 13 int ImageToVideo(char* outDir,char* videoName,char* inputDir,int startFrame,int endFrame, int imgW, 14 int imgH,char* imgExt,double fps,int isColor,int fourcc);//图片序列转化为视频 15 int VideoToImage(char* videoName,char* outDir,char* imgExt,int maxFrameCount);//视频转化为图片序列 16 17 18 int main(int argc, char** argv) 19 { 20 //图片转视频F:\carrdisp_test\origin 21 char* inputDir="F:\\carrdisp_test1\\colorfore\\"; 22 char* videoName="carrdisp_clofr.avi"; 23 char* outDir="F:\\carrdisp_test1\\colorfore\\"; 24 int frames=ImageToVideo(outDir,videoName,inputDir,2,308,448,336,"_Fr.jpg",24,1,CV_FOURCC('X','V','I','D')); 25 std::cout<<"total frames "<<frames<<" have been write to video."<<std::endl; 26 int p; 27 std::cin>>p; 28 29 return 0; 30 } 31 32 //将图片序列转换为视频,返回视频帧数 33 int ImageToVideo(char* outDir,char* videoName,char* inputDir,int startFrame,int endFrame, int imgW,int imgH,char* imgExt,double fps=24,int isColor=1,int fourcc=CV_FOURCC('X','V','I','D')) 34 { 35 //判断输入文件夹是否存在 36 if (_access(inputDir,0)==-1) 37 { 38 std::cout<<"the input directory does not exist!"<<std::endl; 39 return 0; 40 } 41 //判断输出文件夹是否创建 若没有则创建；若为NULL则默认当前工作目录 42 char fullVideoName[255];//输出视频的完整文件名：路径+文件名 43 strcpy_s(fullVideoName,""); 44 if (outDir==NULL) 45 { 46 sprintf_s(fullVideoName,"%s",videoName);//把videoName打印成一个字符串保存在fullVideoName 中 47 } 48 else 49 { 50 if (_access(outDir,0)==-1) 51 { 52 recursive_mkdir(outDir); 53 } 54 sprintf_s(fullVideoName,"%s%s",outDir,videoName);//将字符串outDir和videoName连接起来，打印，保存在fullVideoName中 55 } 56 int frameCount=0; 57 CvVideoWriter *pWriter=NULL; 58 CvSize size=cvSize(imgW,imgH); 59 pWriter=cvCreateVideoWriter(videoName,fourcc,fps,size,isColor);//CREATE WRITER 60 61 IplImage *pImg=NULL; 62 char cur_fn[255];//表示某张图片的路径 63 while(startFrame<=endFrame) 64 { 65 strcpy_s(cur_fn,""); 66 sprintf_s(cur_fn,"%s%d%s",inputDir,startFrame,imgExt);//need to change 67 pImg=cvLoadImage(cur_fn,isColor); 68 if (!pImg) 69 { 70 std::cout<<"can't open an image file"<<std::endl; 71 return frameCount; 72 } 73 cvWriteFrame(pWriter,pImg); 74 cvWaitKey(1); 75 std::cout<<"Write frame "<<startFrame<<std::endl; 76 startFrame++; 77 cvReleaseImage(&pImg); 78 frameCount++; 79 } 80 cvReleaseVideoWriter(&pWriter); 81 rename(videoName,fullVideoName);//移动文件到指定文件夹 82 return frameCount; 83 } 84 //将视频转换为图片序列 返回由视频分解得到的图片总帧数 目前OpenCV只支持AVI格式 因此使用之前需要 85 //将视频转化问AVI格式 86 87 int VideoToImage(char* videoName,char* outDir,char* imgExt,int maxFrameCount) 88 { 89 CvCapture *cap= cvCaptureFromFile(videoName); 90 if (cap==NULL) 91 { 92 return 0; 93 } 94 //保存图片的文件夹路径一定要有，因为OpenCV不会自动创建文件夹 95 if (_access(outDir,0)==-1) 96 { 97 recursive_mkdir(outDir); 98 std::cout<<"the ouput directory does not exist, and the have been created autonomously!"<<std::endl; 99 } 100 char cur_fn[255];//保存当前帧所得图片的文件名 101 IplImage* pImg=NULL; 102 int frame=0; 103 while((pImg=cvQueryFrame(cap))!=NULL&&(frame<maxFrameCount)) 104 { 105 frame++; 106 strcpy_s(cur_fn,""); 107 sprintf_s(cur_fn,"%s%d%s",outDir,frame,imgExt);//这里的设置适合形如 123.jpg 124.jpg的图片序列 108 cvSaveImage(cur_fn,pImg,NULL); 109 } 110 cvReleaseImage(&pImg); 111 cvReleaseCapture(&cap); 112 return frame; 113 } 114 115 116 //该函数借鉴了网上资料，自动创建多级目录 117 int recursive_mkdir( char *dir ) 118 { 119 //分解路径名E:\\AA\\BB\\CC\\ 120 // 121 std::string str = dir; 122 int index = 0; 123 int i = 0; 124 while(1) 125 { 126 std::string::size_type pos = str.find("\\",index); 127 std::string str1; 128 str1 = str.substr(0,pos); 129 if( pos != -1 && i > 0 ) 130 { 131 if (_access(str1.c_str(),0)==-1) 132 { 133 _mkdir(str1.c_str()); 134 } 135 } 136 if( pos==-1 ) 137 { 138 break; 139 } 140 i ++; 141 index = pos+1; 142 } 143 return 0; 144 }
上面程序在VS2010+opencv2.3.1+windows XP环境下测试通过，使用时注意设置好开始帧，结束帧，图片格式即可。本人图片目录如：
F:\carrdisp_test1\colorfore\2_Fr.jpg
其中只调用了图片序列写为视频的函数，另一个函数调用方法相似，如下举例：
View Code
//视频转图片 char* videoName1="E:\\VIDEOS\\TEST\\Videos\\woman.avi"; char* outDir1="E:\\VIDEOS\\TEST\\Sequences\\WoemenSequence\\"; int images=VideoToImage(videoName1,outDir1,".jpg",200); std::cout<<"total frames have been extracted from video."<<std::endl; int p; std::cin>>p;
