目标检测——CodeBook 算法测试代码

// picturepreprocess.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <fstream>

#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/core/core.hpp>
#include <opencv2/video/background_segm.hpp>
#include <opencv2/legacy/legacy.hpp>
#include <cvaux.h>

using namespace std;
using namespace cv;

//const int NCHANNELS = 3;
//bool ch[NCHANNELS]={true,true,true};

int _tmain(int argc, _TCHAR* argv[])
{
//CvScalar p;
//ofstream outfile("C:\\Users\\admin\\Desktop\\gray\\gray.txt");
IplImage* rawImage = 0, *yuvImage = 0;
IplImage *ImaskCodeBook = 0,*ImaskCodeBookCC = 0;
CvCapture* capture = 0;

int nframes = 0;
int nframesToLearnBG =200;//训练帧数选择
CvBGCodeBookModel* model = 0;
model = cvCreateBGCodeBookModel(); //codebook方法参数初始化（选择默认值）
model->modMin[0] = 3;
model->modMin[1] = model->modMin[2] = 3;
model->modMax[0] = 10;
model->modMax[1] = model->modMax[2] = 10;
model->cbBounds[0] = model->cbBounds[1] = model->cbBounds[2] = 10;

bool pause = false;

cvNamedWindow( "Raw", 1 );
cvNamedWindow( "Foreground",1);
cvNamedWindow( "Foreground_ConnectComp",1);
capture =cvCreateFileCapture(".....");//初始化从文件中获取视频
while(rawImage = cvQueryFrame( capture ))
{
nframes++;
if(nframes == 1)
{
yuvImage = cvCloneImage(rawImage);
ImaskCodeBook = cvCreateImage(cvGetSize(rawImage), IPL_DEPTH_8U, 1 );
ImaskCodeBookCC = cvCreateImage(cvGetSize(rawImage), IPL_DEPTH_8U, 1 );
cvSet(ImaskCodeBook,cvScalar(255));

}
else
{
cvCvtColor(rawImage, yuvImage, CV_BGR2YCrCb);
if(!pause && nframes-1 < nframesToLearnBG)
cvBGCodeBookUpdate( model, yuvImage );
if(nframes-1 == nframesToLearnBG)
cvBGCodeBookClearStale( model, model->t/2 );

if( nframes-1 >= nframesToLearnBG){
// 提取前景
cvBGCodeBookDiff(model, yuvImage, ImaskCodeBook);//codebook方法中，背景减除
// 对前景做连通域分割
cvCopy(ImaskCodeBook,ImaskCodeBookCC);
cvSegmentFGMask(ImaskCodeBookCC);
}

cvShowImage("Raw", rawImage);
cvShowImage("Foreground",ImaskCodeBook);

//查看像素
//for (int i=0;i<ImaskCodeBook->width;i++)
//{
//for (int j=0;j<ImaskCodeBook->height;j++)
//{

// p=cvGet2D(ImaskCodeBook,j,i);

//outfile<<p.val[0]<<"";
//}
//outfile<<endl;
//}

cvShowImage("Foreground_ConnectComp",ImaskCodeBookCC);
char c=cvWaitKey(30);
if(c==27) //按下Esc键退出视频播放
break;
}

}

cvReleaseCapture( &capture );
cvDestroyWindow( "Raw" );
cvDestroyWindow( "Foreground");
cvDestroyWindow( "Foreground_ConnectComp");

}