OpenCV

开发平台：

Visual C++

myfgdetector.cpp：源码内容

							#include "myfgdetector.h"
MyFGDetector::MyFGDetector()
{
  this->Create(0.003, 60.0);
}
MyFGDetector::MyFGDetector(double alpha, double threshold)
{
  this->Create(alpha, threshold);
}
MyFGDetector::~MyFGDetector()
{
  this->Release();
}
void MyFGDetector::Create(double alpha, double threshold)
{
  m_framecount = 0;
  m_alpha = alpha;
  m_threshold = threshold;
  m_pFrame=0;
  m_pFGMask=0;
  m_pFrameMat=0;
  m_pFGMat=0;
  m_pBkMat=0;
}
IplImage* MyFGDetector::GetMask()
{
  return m_pFGMask;
}
void MyFGDetector::Release()
{
  m_framecount = 0;
  if(m_pFrame) cvReleaseImage(&m_pFrame);
  if(m_pFGMask) cvReleaseImage(&m_pFGMask);
  if(m_pFrameMat) cvReleaseMat(&m_pFrameMat);
  if(m_pFGMat) cvReleaseMat(&m_pFGMat);
  if(m_pBkMat) cvReleaseMat(&m_pBkMat);
}
void MyFGDetector::Process(IplImage* pFrame)
{
  if(m_framecount==0)
    {
      m_pFGMask = cvCreateImage(cvSize(pFrame->width, pFrame->height),  IPL_DEPTH_8U,1);
      m_pBkMat = cvCreateMat(pFrame->height, pFrame->width, CV_32FC1);
      m_pFGMat = cvCreateMat(pFrame->height, pFrame->width, CV_32FC1);
      m_pFrameMat = cvCreateMat(pFrame->height, pFrame->width, CV_32FC1);
      
      if( !m_pFGMask || !m_pBkMat || !m_pFGMat || !m_pFrameMat)
	{
	  fprintf(stderr, "Can not alloc memeory.n");
	  return ;
	}
      //convert to single channel
      cvCvtColor(pFrame, m_pFGMask, CV_BGR2GRAY);
      cvConvert(m_pFGMask, m_pFrameMat);
      cvConvert(m_pFGMask, m_pFGMat);
      cvConvert(m_pFGMask, m_pBkMat);
    }
  //转化成单信道图像再处理
  cvCvtColor(pFrame, m_pFGMask, CV_BGR2GRAY);
  cvConvert(m_pFGMask, m_pFrameMat);
  //高斯滤波先，以平滑图像
  cvSmooth(m_pFrameMat, m_pFrameMat, CV_GAUSSIAN, 3, 0, 0);
	  
  //当前帧跟背景图相减
  cvAbsDiff(m_pFrameMat, m_pBkMat, m_pFGMat);
  //二值化前景图
  cvThreshold(m_pFGMat, m_pFGMask, m_threshold, 255.0, CV_THRESH_BINARY);
  //进行形态学滤波，去掉噪音  
  cvErode(m_pFGMask, m_pFGMask, 0, 1);
  cvDilate(m_pFGMask, m_pFGMask, 0, 1);
  //更新背景
  cvRunningAvg(m_pFrameMat, m_pBkMat, 0.003, 0);
 
  m_framecount++;
}

						