OpenCV读视频文件和运动问题检测

下面这个程序 来自于仕琪的讲稿 《使用OpenCV进行图像处理》]中的例程：

cvRunningAvg

　　opencv中的函数，用来更新移动平均。用法：

　　void cvRunningAvg(const CvArr * image,

　　CvArr* acc,

　　double alpha,

　　const CvArr* mask=NULL)

　　image:输入图像，1或3通道，8比特或32比特的float型

　　acc:累加器，和image一样大小

　　alpha：更新时，image所占的权重

　　mask：操作符掩码

　　if mask(x,y)!= 0 (1-alpha)*acc(x,y)+alpha*image(x,y) =>acc(x,y)

[cpp]  view plain copy

1. #if 0  
2. #include <stdio.h>  
3.   
4. #include <cv.h>  
5. #include <cxcore.h>  
6. #include <highgui.h>  
7.   
8. #pragma comment(lib, "cv.lib")  
9. #pragma comment(lib, "cxcore.lib")  
10. #pragma comment(lib, "highgui.lib")  
11.   
12. int main( int argc, char** argv )  
13. {  
14.     //声明IplImage指针  
15.     IplImage* pFrame = NULL;   
16.     IplImage* pFrImg = NULL;  
17.     IplImage* pBkImg = NULL;  
18.   
19.     CvMat* pFrameMat = NULL;  
20.     CvMat* pFrMat = NULL;  
21.     CvMat* pBkMat = NULL;  
22.   
23.     CvCapture* pCapture = NULL;  
24.   
25.     int nFrmNum = 0;  
26.   
27.     //创建窗口  
28.     cvNamedWindow("video", 1);  
29.     cvNamedWindow("background",1);  
30.     cvNamedWindow("foreground",1);  
31.     //使窗口有序排列  
32.     cvMoveWindow("video", 30, 0);  
33.     cvMoveWindow("background", 360, 0);  
34.     cvMoveWindow("foreground", 690, 0);  
35.   
36.   
37.     if( !(pCapture = cvCaptureFromAVI(".\\test.avi")))  
38.     {  
39.         //pCapture = cvCaptureFromCAM(-1))  
40.         fprintf(stderr, "Can not open camera.\n");  
41.         return -2;  
42.     }  
43.   
44.   
45.     //逐帧读取视频  
46.     while(pFrame = cvQueryFrame( pCapture ))  
47.     {  
48.         nFrmNum++;  
49.   
50.         //如果是第一帧，需要申请内存，并初始化  
51.         if(nFrmNum == 1)  
52.         {  
53.             pBkImg = cvCreateImage(cvSize(pFrame->width, pFrame->height),  IPL_DEPTH_8U,1);  
54.             pFrImg = cvCreateImage(cvSize(pFrame->width, pFrame->height),  IPL_DEPTH_8U,1);  
55.   
56.             pBkMat = cvCreateMat(pFrame->height, pFrame->width, CV_32FC1);  
57.             pFrMat = cvCreateMat(pFrame->height, pFrame->width, CV_32FC1);  
58.             pFrameMat = cvCreateMat(pFrame->height, pFrame->width, CV_32FC1);  
59.   
60.             //转化成单通道图像再处理  
61.             cvCvtColor(pFrame, pBkImg, CV_BGR2GRAY);  
62.             cvCvtColor(pFrame, pFrImg, CV_BGR2GRAY);  
63.   
64.             cvConvert(pFrImg, pFrameMat);  
65.             cvConvert(pFrImg, pFrMat);  
66.             cvConvert(pFrImg, pBkMat);  
67.         }  
68.         else  
69.         {  
70.             cvCvtColor(pFrame, pFrImg, CV_BGR2GRAY);  
71.             cvConvert(pFrImg, pFrameMat);  
72.             //高斯滤波先，以平滑图像  
73.             //cvSmooth(pFrameMat, pFrameMat, CV_GAUSSIAN, 3, 0, 0);  
74.   
75.             //当前帧跟背景图相减  
76.             cvAbsDiff(pFrameMat, pBkMat, pFrMat);  
77.   
78.             //二值化前景图  
79.             cvThreshold(pFrMat, pFrImg, 60, 255.0, CV_THRESH_BINARY);  
80.   
81.             //进行形态学滤波，去掉噪音    
82.             //cvErode(pFrImg, pFrImg, 0, 1);  
83.             //cvDilate(pFrImg, pFrImg, 0, 1);  
84.   
85.             //更新背景  
86.             cvRunningAvg(pFrameMat, pBkMat, 0.003, 0);  
87.             //将背景转化为图像格式，用以显示  
88.             cvConvert(pBkMat, pBkImg);  
89.   
90.             //显示图像  
91.             cvShowImage("video", pFrame);  
92.             cvShowImage("background", pBkImg);  
93.             cvShowImage("foreground", pFrImg);  
94.   
95.             //如果有按键事件，则跳出循环  
96.             //此等待也为cvShowImage函数提供时间完成显示  
97.             //等待时间可以根据CPU速度调整  
98.             if( cvWaitKey(20) >= 0 )  
99.             {  
100.                 break;  
101.             }  
102.         }  
103.     }  
104.     cvWaitKey();  
105.   
106.     //销毁窗口  
107.     cvDestroyWindow("video");  
108.     cvDestroyWindow("background");  
109.     cvDestroyWindow("foreground");  
110.   
111.     //释放图像和矩阵  
112.     cvReleaseImage(&pFrImg);  
113.     cvReleaseImage(&pBkImg);  
114.   
115.     cvReleaseMat(&pFrameMat);  
116.     cvReleaseMat(&pFrMat);  
117.     cvReleaseMat(&pBkMat);  
118.   
119.     cvReleaseCapture(&pCapture);  
120.   
121.     return 0;  
122. }  
123. #else  
124. #include <cv.h>  
125. #include <highgui.h>  
126.   
127. #pragma comment(lib, "cv.lib")  
128. #pragma comment(lib, "cxcore.lib")  
129. #pragma comment(lib, "highgui.lib")  
130.   
131. int main( int argc, char** argv )  
132. {  
133.     CvCapture* capture = NULL;  
134.     IplImage *frame = NULL;  
135.   
136.     IplImage  *frame_last = NULL;//存储上一帧  
137.     IplImage  *m_out_image = NULL;  
138.   
139.     IplImage* frame_gray = NULL;  
140.     IplImage* frame_gray_last = NULL;  
141.     IplImage* m_out_image_copy = NULL;  
142.     int nFrmNum = 0;  
143.   
144.     //capture = cvCaptureFromCAM(-1);   
145.   
146.     capture = cvCaptureFromAVI(".\\test.avi");  
147.   
148.     //创建窗口  
149.     cvNamedWindow( "result", 1 );  
150.     cvNamedWindow("差分");  
151.   
152.     //逐帧读取视频  
153.     while(frame = cvQueryFrame( capture ))  
154.     {  
155.         nFrmNum++;  
156.   
157.         //如果是第一帧，需要申请内存，并初始化  
158.         if(nFrmNum == 1)  
159.         {  
160.             frame_last = cvCreateImage(cvSize(frame->width,frame->height),IPL_DEPTH_8U, frame->nChannels);  
161.             cvCopy(frame,frame_last);  
162.         }  
163.         else  
164.         {  
165.             frame_gray = cvCreateImage(cvSize(frame->width,frame->height),IPL_DEPTH_8U, 1);  
166.             frame_gray_last = cvCreateImage(cvSize(frame->width,frame->height),IPL_DEPTH_8U, 1);  
167.   
168.             m_out_image = cvCreateImage(cvSize(frame->width,frame->height),IPL_DEPTH_8U, 1);  
169.             m_out_image_copy = cvCreateImage( cvSize(frame->width,frame->height),IPL_DEPTH_8U, 1 );  
170.   
171.             cvCvtColor(frame,frame_gray,CV_RGB2GRAY);  
172.   
173.             cvCvtColor(frame_last,frame_gray_last,CV_RGB2GRAY);  
174.   
175.             cvAbsDiff(frame_gray,frame_gray_last,m_out_image);  
176.   
177.             //cvThreshold(m_out_image,m_out_image,128, 255,CV_THRESH_BINARY);  
178.   
179.             //将处理后的图像写出到第二个图像中  
180.   
181.             if( m_out_image_copy->origin == IPL_ORIGIN_TL )  
182.             {  
183.                 cvCopy( m_out_image, m_out_image_copy, 0 );  
184.             }  
185.             else  
186.             {  
187.                 cvFlip( m_out_image, m_out_image_copy, 0 );  
188.             }  
189.   
190.             frame_last = cvCreateImage(cvSize(frame->width,frame->height),IPL_DEPTH_8U, frame->nChannels);  
191.             cvCopy(frame,frame_last);  
192.   
193.             cvShowImage("差分",m_out_image_copy);  
194.             cvShowImage("result",frame);  
195.   
196.             cvReleaseImage( &m_out_image_copy );  
197.             cvReleaseImage( &frame_gray_last );  
198.             cvReleaseImage( &frame_gray );  
199.   
200.             if( cvWaitKey( 20 ) >= 0 )  
201.             {  
202.                 break;  
203.             }  
204.         }  
205.     }  
206.     cvWaitKey();  
207.   
208.     cvReleaseImage( &m_out_image );  
209.     cvReleaseImage( &frame_last );  
210.     cvReleaseImage( &frame );  
211.   
212.     cvReleaseCapture( &capture );  
213.   
214.     cvDestroyWindow("result");  
215.     cvDestroyWindow("差分");  
216.   
217.     return 0;  
218. }  
219. #endif