vibe背景建模算法的实现

// vibe.cpp : Defines the entry point for the console application. // // lx_Picture.cpp : Defines the entry point for the console application. // #include <stdio.h> #include "highgui.h" #include "hanshu.h" #include "cv.h" #include <stdlib.h> #include <cxcore.h> //#include <ctime> #include <time.h> int nFrmNum = 0;//记录帧数 int Width;//记录帧的宽度 int Height;//记录帧的高度 int FrameRate;//记录视频帧率 int main(int argc, char* argv[]) { IplImage* pFrame=NULL;CvMat* pFrameMat = NULL;//pFrame对象 IplImage* pAfter=NULL;CvMat* pAfterMat=NULL;//保存pFrame对应的灰度图像 IplImage* segMap=NULL;CvMat* segMat=NULL;//保存处理后的信息 double fps; CvSize size;CvVideoWriter* Save_result=NULL; clock_t start; clock_t end; IplImage* dstImg=NULL; //创建一个随机数生成器 CvRNG rng; //定义字体 CvFont font; //要读取的视频文件和保存的视频文件路径 char* openfile="D:\\CFiles\\Vibe\\Capture_20130924.avi"; char* outfile="D:\\CFiles\\Vibe\\vibe.avi"; char filename[100];//保存的图像位置和名称 //打开视频文件 CvCapture* pCapture=cvCreateFileCapture(openfile); if(pCapture==NULL) { printf("video file open error!"); return -1; } //获取视频相关信息，帧率和大小 fps=cvGetCaptureProperty(pCapture,CV_CAP_PROP_FPS); size=cvSize((int)cvGetCaptureProperty(pCapture,CV_CAP_PROP_FRAME_WIDTH)*2, (int)cvGetCaptureProperty(pCapture,CV_CAP_PROP_FRAME_HEIGHT)); //创建输出视频文件 Save_result=cvCreateVideoWriter(outfile,CV_FOURCC('X','V','I','D'),fps,size,1); dstImg=cvCreateImage(size,IPL_DEPTH_8U,3);//创建要保存的图像 //创建窗口 cvNamedWindow("video",CV_WINDOW_AUTOSIZE); cvInitFont( &font, CV_FONT_HERSHEY_COMPLEX_SMALL ,1, 1, 0, 1, 8); //逐帧读取视频并进行处理 while(pFrame = cvQueryFrame( pCapture )){ nFrmNum++; //如果是第一帧，申请内存并进行初始化 if(nFrmNum==1){ segMap = cvCreateImage(cvSize(pFrame->width, pFrame->height), IPL_DEPTH_8U,1); segMat = cvCreateMat(pFrame->height, pFrame->width, CV_32FC1); pAfter=cvCreateImage(cvSize(pFrame->width, pFrame->height), IPL_DEPTH_8U,1); pAfterMat=cvCreateMat(pFrame->height, pFrame->width, CV_32FC1); //转化成单通道图像再处理 cvCvtColor(pFrame, pAfter, CV_BGR2GRAY); cvConvert(pAfter, pAfterMat); // Initialize(pAfterMat,rng); } else { cvCvtColor(pFrame,pAfter,CV_BGR2GRAY); cvConvert(pAfter,pAfterMat); start=clock(); update(pAfterMat,segMat,rng,nFrmNum);//更新背景 end = clock(); printf("The running time is %f\n",(double)(end-start)/CLOCKS_PER_SEC); cvConvert(segMat,segMap); //载入原图像到目标图像 cvSetImageROI(dstImg, cvRect(0, 0, pFrame->width, pFrame->height)); cvCopy(pFrame, dstImg,NULL); cvResetImageROI(dstImg); //将segMap转换成三通道图像存在pFrame中 cvCvtColor(segMap,pFrame,CV_GRAY2BGR); //载入检测后的图像到目标图像 cvSetImageROI(dstImg, cvRect(pFrame->width, 0, pFrame->width*2, pFrame->height)); cvCopy(pFrame, dstImg,NULL); cvResetImageROI(dstImg); //显示提示文字 cvPutText(dstImg, "Origin Video", cvPoint(0,pFrame->height-5), &font,CV_RGB(255,255,0)); cvPutText(dstImg, "ViBe Video", cvPoint(pFrame->width,pFrame->height-5),&font,CV_RGB(255,255,0)); //dstImg->origin=1; //保存视频和输出 dstImg->origin=1; cvWriteFrame(Save_result,dstImg); /*输出图片 if(nFrmNum<11) sprintf(filename,"E:\\Result\\Vibe\\AVSS_PV_Easy_Divx_Xvid\\000%d_Vibe.jpg",nFrmNum-1); else if(nFrmNum<101) sprintf(filename,"E:\\Result\\Vibe\\AVSS_PV_Easy_Divx_Xvid\\00%d_Vibe.jpg",nFrmNum-1); else if(nFrmNum<1001) sprintf(filename,"E:\\Result\\Vibe\\AVSS_PV_Easy_Divx_Xvid\\0%d_Vibe.jpg",nFrmNum-1); else if(nFrmNum<10001) sprintf(filename,"E:\\Result\\Vibe\\AVSS_PV_Easy_Divx_Xvid\\%d_Vibe.jpg",nFrmNum-1); cvSaveImage(filename,dstImg); */ cvShowImage("video",dstImg); if(cvWaitKey(5)>=0) break; } } cvReleaseImage(&pFrame);cvReleaseMat(&pFrameMat); cvReleaseImage(&pAfter);cvReleaseMat(&pAfterMat); cvReleaseImage(&segMap);cvReleaseMat(&segMat); cvReleaseVideoWriter(&Save_result); cvReleaseImage(&dstImg); cvDestroyWindow("video"); return 0; }