背景差法OpenCV实现程序代码

背景差法OpenCV实现程序代码

默认分类 2010-11-12 22:10:27 阅读70 评论0   字号：大中小 订阅

#include "highgui.h" #include "cv.h" #include <stdio.h> #include"例子一.h" int main(int argc,char**argv) { //声明指针，矩阵 IplImage*image=NULL; IplImage*backimage=NULL; IplImage*foreimage=NULL; IplImage*tank=NULL;   CvMat*imagemat=NULL; CvMat*foremat=NULL; CvMat*backmat=NULL; CvCapture*capture=NULL; int num=0; //创建窗口 cvNamedWindow("cvcamwindow",CV_WINDOW_AUTOSIZE); cvNamedWindow("foreground",1); cvNamedWindow("background",1); cvNamedWindow("wo",1); //读取摄像头        if(argc==1||(argc==2&&strlen(argv[1])==1&&isdigit(argv[1][0])))               capture=cvCaptureFromCAM(argc==2?argv[1][0]-'0':0);        else if(argc==2)               capture=cvCaptureFromAVI(argv[1]);     //声明视频结构 int isColor = 1;        int frameW=160;        int frameH=120;        int fps=8;    CvVideoWriter* camWriter=cvCreateVideoWriter("E://directx//jianhua12.avi",-1,fps,cvSize(frameW,frameH),isColor);        if(!capture)        {               fprintf(stderr,"could not initialize capturing.../n");           return 0;       }   //处理图像        while(image=cvQueryFrame(capture))        {               num++;               if(num==1)//第一针申请内存               { foreimage=cvCreateImage(cvGetSize(image),8,1);                   backimage=cvCreateImage(cvGetSize(image),8,1);            tank=cvCreateImage(cvGetSize(image),8,3);               backmat=cvCreateMat(image->height,image->width,CV_32FC1);             foremat=cvCreateMat(image->height,image->width,CV_32FC1);             imagemat=cvCreateMat(image->height,image->width,CV_32FC1);                  //转换成单通道处理                cvCvtColor(image,backimage,CV_BGR2GRAY);                cvCvtColor(image,foreimage,CV_BGR2GRAY);             //矩阵初始化             cvConvert(foreimage,foremat);             cvConvert(foreimage,backmat);             cvConvert(foreimage,imagemat);           }               else               {                      cvCvtColor(image,foreimage,CV_BGR2GRAY);                      cvConvert(foreimage,imagemat);                                          //高斯滤波，平滑图像                      cvSmooth(imagemat,imagemat,CV_GAUSSIAN,3,0,0);                      //当前跟背景差分                      cvAbsDiff(imagemat,backmat,foremat);                      cvConvert(foremat,foreimage);                     //二值化                      cvThreshold(foreimage,foreimage,60,255.0,CV_THRESH_BINARY);                      //进行形态学滤波，去掉噪音                      cvDilate(foreimage, foreimage, 0, 1);                     cvErode(foreimage, foreimage, 0, 1);                      cvErode(foreimage, foreimage, 0, 1);                      cvDilate(foreimage, foreimage, 0, 1);

）//还原处理
                     int nl=foreimage->height;
                     int nc=foreimage->width;
                   BwImage foreimageA(foreimage);
                     RgbImage imageA(image);
                     RgbImage tankA(tank);
                     for(int i=0;i<nl;i++)
                     {
                            for(int j=0;j<nc;j++)
                            {
                                if (foreimageA[i][j]==0)
                   {
tankA[i][j].b=0;
                                      tankA[i][j].g=0;
                                      tankA[i][j].r=0;
}
                  else      
tankA[i][j]=imageA[i][j];                               
                            }
                     }
           tank->origin=image->origin;
//保存前景
                     cvWriteFrame(camWriter,tank);
                     //更新背景
                     cvRunningAvg(imagemat,backmat,0.005,0);
                  //将背景转化为图像格式
                     cvConvert(backmat,backimage);
                     //使图像正立
                     backimage->origin=image->origin;
                     foreimage->origin=image->origin;
                     //显示图像
                     cvShowImage("cvcamwindow", image);
           cvShowImage("background", backimage);
           cvShowImage("foreground" ,foreimage);
                     cvShowImage("wo",tank);
                  //时间缓冲
           int key=cvWaitKey(90);
                     if(cvWaitKey(2)>=0)
                            break;
              }
       }
 
       cvDestroyWindow("cvcam window");
      cvDestroyWindow("background");
      cvDestroyWindow("foreground");
      cvDestroyWindow("wo");
 
       cvReleaseImage(&foreimage);
      cvReleaseImage(&backimage);
      cvReleaseImage(&tank);
      
       cvReleaseMat(&imagemat);
      cvReleaseMat(&foremat);
      cvReleaseMat(&backmat);
 
     cvReleaseVideoWriter (&camWriter);
    
      return 0;
}