OPENCV的高斯背景提取.docx

1. /*2. *3. *基于openCV的背景剪除方法,高斯模型4. */5. #include 6. 7. #include 8. #include 9. #include 10. #include 11. 12. intmain(intargc,char*argv)13. 14. 15. 16. #if0/作测试用，当不需要手动修改输入文件名时，可以将1改为0 17. argc=2;18. argv0=3.exe;/生成的可执行文件名，必须有！ 19. argv1=capfile3.avi;20. #endif 21. 22. 23. /声明IplImage指针 24. IplImage*tmp_frame=NULL;25. 26. CvMemStorage*m_storage=cvCreateMemStorage(0);27. CvSeq*m_contour=NULL;28. IplImage*dst=0;29. 30. CvCapture*m_capture=NULL;31. 32. if(argc2)33. 34. fprintf(stderr,Usage:bkgrdvideo_file_namen);35. return-1;36. 37. 38. /打开摄像头 39. if(argc=1)40. 41. if(!(m_capture=cvCaptureFromCAM(-1)42. 43. fprintf(stderr,Cannotopencamera.n);44. return-2;45. 46. 47. 48. /打开视频文件 49. if(argc=2)50. 51. if(!(m_capture=cvCaptureFromFile(argv1)52. 53. fprintf(stderr,Cannotopenvideofile%sn,argv1);54. return-2;55. 56. 57. 58. 59. tmp_frame=cvQueryFrame(m_capture);60. 61. 62. dst=cvCreateImage(cvGetSize(tmp_frame),8,1);63. dst-origin=1;64. 65. cvCvtColor(tmp_frame,dst,CV_BGR2GRAY);66. /cvCvtColor(pFrame,pFrImg,CV_BGR2GRAY); 67. 68. if(!tmp_frame)69. 70. fprintf(stderr,读取视频文件失败，请重新打开真彩色(24位)视频!.n);71. /MessageBox(读取视频文件失败，请重新打开真彩色(24位)视频!,错误信息); 72. return-1;73. 74. 75. cvNamedWindow(video,1);76. cvNamedWindow(背景图像,1);77. cvNamedWindow(前景图像,1);78. 79. cvMoveWindow(video,30,0);80. cvMoveWindow(背景图像,360,0);81. cvMoveWindow(前景图像,690,0);82. 83. 84. /* 85. IplImage*dst1=NULL;86. dst1=cvCreateImage(cvGetSize(tmp_frame),IPL_DEPTH_8U,1);87. dst1-origin=1;88. cvCvtColor(tmp_frame,dst1,CV_BGR2GRAY);89. 90. 91. /创建多高斯模型 92. CvGaussBGStatModelParamsparams;93. params.win_size=200;/1/alpha/影响背景 94. params.bg_threshold=0.90;/CV_BGFG_MOG_BACKGROUND_THRESHOLD;/前B个权重w的和/大点好 95. params.std_threshold=4.0;/CV_BGFG_MOG_STD_THRESHOLD;/sigma 96. params.weight_init=0.005;/CV_BGFG_MOG_WEIGHT_INIT;/初始权重w 97. params.variance_init=15*15;/CV_BGFG_MOG_SIGMA_INIT*CV_BGFG_MOG_SIGMA_INIT;/方差 98. params.minArea=30;/CV_BGFG_MOG_MINAREA 99. params.n_gauss=5;/CV_BGFG_MOG_NGAUSSIANS;/高斯模型数量 100. 101. CvBGStatModel*bg_model=cvCreateGaussianBGModel(dst1,ms);102. 103. 104. 105. 106. for(intfr=1;tmp_frame;tmp_frame=cvQueryFrame(m_capture),fr+)107. 108. /printf(frame#%d:,fr); 109. /获得背景模型更新过程所花费的时间，以CPU时钟/每微秒为单位计数 110. /doublet=(double)cvGetTickCount();/模型更新之前 111. 112. 113. /* 114. cvCvtColor(tmp_frame,dst1,CV_BGR2GRAY);115. 116. cvUpdateBGStatModel(dst1,bg_model);117. /真正的函数实现体是在:icvUpdateFGDStatModel(tmp_frame,bg_model) 118. 119. /t=(double)cvGetTickCount()-t;/模型更新之后 120. 121. /以左下角为坐标原点 122. bg_model-foreground-origin=bg_model-background-origin=1;123. 124. /cvErode(bg_model-background,bg_model-background,0,1); 125. cvErode(bg_model-foreground,bg_model-foreground,0,1);126. /滤除噪声 127. cvDilate(bg_model-foreground,bg_model-foreground,0,1);128. /cvDilate(bg_model-background,bg_model-background,0,1); 129. /滤除噪声 130. 131. /printf(%.1f毫秒n,t/(cvGetTickFrequency()*1000.); 132. /最后输出结果以毫秒为单位 133. 134. cvShowImage(video,tmp_frame);135. cvShowImage(背景图像,bg_model-background);136. cvCopy(bg_model-foreground,dst);137. cvSmooth(dst,dst,CV_GAUSSIAN,5);138. cvMorphologyEx(dst,dst,0,0,CV_MOP_CLOSE,3);139. cvMorphologyEx(dst,dst,0,0,CV_MOP_OPEN,1);140. 141. /提取轮廓到contour序列中 142. 143. /cvFindContours仅能处理单通道、颜色深度为8位的图像的轮廓提取 144. cvFindContours(dst,m_storage,&m_contour,sizeof(CvContour),145. CV_RETR_LIST,CV_CHAIN_APPROX_SIMPLE,cvPoint(0,0);146. 147. for(;m_contour!=0;m_contour=m_contour-h_next)148. 149. CvScalarcolor=CV_RGB(255,0,255);150. CvScalarcolor_rect=CV_RGB(0,255,255);151. 152. CvRectm_contour_rect=cvBoundingRect(m_contour,1);153. 154. 155. if(fabs(cvContourArea(m_contour)400.0)156. 157. cvRectangle(dst,cvPoint(m_contour_rect.x,m_contour_rect.y),158. cvPoint(m_contour_rect.x+m_contour_rect.width),(m_contour_rect.y+m_contour_rect.height),159. color_rect,2,8,0);160. 161. /*replaceCV_FILLEDwith1toseetheoutlines*/162. cvDrawContours(dst,m_contour,color,color,-1,CV_FILLED,8,cvPoint(0,0);163. cvShowImage(前景图像,dst);164. 165. 166. 167. intk=cvWaitKey(10);/等待10毫秒 168. if(k=q)break;169. 170. 171. c