OpenCV 视频与图片序列相互转换

  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 }

  //视频转图片
   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;