/**

* @file main-opencv.cpp

* @date July 2014

* @brief An exemplative main file for the use of ViBe and OpenCV*/

//#include

#include "vibe-background-sequential.h"

using namespacecv;using namespacestd;const int minArea = 2; //舍去面积极小的方框

const double maxAreaRatio = 0.1; //舍去面积极大的方框

const double boxRatio = 0.1; //舍去长宽比严重失衡的方框

/**

* Displays instructions on how to use this program.*/

voidhelp()

{

cout<< "--------------------------------------------------------------------------" <" <

}void processVideo(char*videoFilename);void contour(const Mat &mor, Mat &img);/**

* Main program. It shows how to use the grayscale version (C1R) and the RGB version (C3R).*/

int main(int argc, char*argv[])

{/*Print help information.*/help();/*Check for the input parameter correctness.*/

/*if (argc != 2) {

cerr <

cerr <

return EXIT_FAILURE;

}

/* Create GUI windows.*/

//namedWindow("Frame");//namedWindow("Segmentation by ViBe");

processVideo("framelink_yd.avi"); //读取 framelink_yd.avi 视频进行处理

/*Destroy GUI windows.*/destroyAllWindows();returnEXIT_SUCCESS;

}/**

* Processes the video. The code of ViBe is included here.

*

* @param videoFilename The name of the input video file.*/

void processVideo(char*videoFilename)

{

VideoCapture capture(videoFilename);/*Create the capture object.*/

if (!capture.isOpened()) {

cerr<< "Unable to open video file:" << videoFilename<< endl; /*Error in opening the video input.*/exit(EXIT_FAILURE);

}

clock_t start, finish;doubletotal;

start=clock();/*Variables.*/

static int frameNumber = 1; /*The current frame number*/

int lastCount = 0;int probFactor = gradient_Factor; /*概率因子，用梯度因子初始化概率因子*/Mat frame,frame1;/*Current frame.*/Mat segmentationMap;/*Will contain the segmentation map. This is the binary output map.*/

int keyboard = 0; /*Input from keyboard. Used to stop the program. Enter 'q' to quit.*/

char fileName[200] = { 0};int sampleCounts[10] = {0};int speSamples[10] = {0};

vibeModel_Sequential_t*model = NULL; /*Model used by ViBe.*/

int heig = 512;int widt = 1340;

Mat res= Mat::zeros(heig, widt, CV_8UC3); //res为三通道像素帧，用来保存最后合并的图像

/*创建保存视频的文件名并打开*/

const string Name = "res.avi";

VideoWriter writer;

Size sz(widt, heig);

writer.open(Name, CV_FOURCC('M', 'J', 'P', 'G'), 70, sz, true);while ((char)keyboard != 'q' && (char)keyboard != 27 ) { /*Read input data. ESC or 'q' for quitting.*/

if (!capture.read(frame1)) { /*Read the current frame.*/cerr<< "Unable to read next frame." <

cerr<< "Exiting..." <

}/*Applying ViBe.

* If you want to use the grayscale version of ViBe (which is much faster!):

* (1) remplace C3R by C1R in this file.

* (2) uncomment the next line (cvtColor).*/cvtColor(frame1, frame, CV_BGR2GRAY);//将三通道图像帧转换为单通道

if ( frameNumber==1) {

segmentationMap=Mat(frame.rows, frame.cols, CV_8UC1);

model= (vibeModel_Sequential_t*)libvibeModel_Sequential_New();

libvibeModel_Sequential_AllocInit_8u_C1R(model, frame.data, frame.cols, frame.rows);

class_samples(model,frame.data, sampleCounts,speSamples,frame.cols, frame.rows);

}

/*ViBe: Segmentation and updating.*/libvibeModel_Sequential_Segmentation_8u_C1R(model, frame.data, segmentationMap.data);

libvibeModel_Sequential_Update_8u_C1R(model, frame.data, segmentationMap.data,&probFactor,frameNumber);

medianBlur(segmentationMap, segmentationMap,3); /*3x3 median filtering*/contour(segmentationMap, frame1); //在原图上框出动目标，至此图像处理完毕//sprintf(fileName, "results55_2/%06d.jpg",frameNumber);//imwrite(fileName, frame1);

Mat segmentationMap1;//新定义一个帧变量，不能 cvtColor(segmentationMap, segmentationMap, CV_GRAY2BGR)，因为 segmentationMap 为单通道帧；

cvtColor(segmentationMap, segmentationMap1, CV_GRAY2BGR); //将单通道帧图像 segmentationMap 转化为三通道segmentationMap1，因为 res 帧为三通道帧图像//segmentationMap1要合并在 res 中；

/*Shows the current frame and the segmentation map.*/

//imshow("Frame", frame1);//imshow("Segmentation by ViBe", segmentationMap);

/*将 segmentationMap1 和 frame1 合并成一帧存放在 res 中*/segmentationMap1.copyTo(res(Rect(0, 0, 640, 512)));

frame1.copyTo(res(Rect(700, 0, 640, 512)));

imshow("res", res); //显示合并后的图像/*将 res 写入打开的视频文件中*/writer<

keyboard= waitKey(1); /*Gets the input from the keyboard.*/}

finish=clock();

total=(double)(finish-start);

cout<

cout<

capture.release();/*Delete capture object.*/libvibeModel_Sequential_Free(model);/*Frees the model.*/}/*框出运动目标*/

void contour(const Mat &mor, Mat &img)

{int img_size = img.cols *img.rows;

Mat tmp= (mor == 255);//Each detected contour is stored as a vector of points

vector >contours;

vector hierarchy; //containing information about the image topology

findContours(tmp, contours, hierarchy, CV_RETR_EXTERNAL, CV_CHAIN_APPROX_SIMPLE);for (size_t k = 0; k < contours.size(); k++) {

Rect rect=boundingRect(contours[k]);double whratio = double(rect.width) / double(rect.height);double hwratio = double(rect.height) / double(rect.width);double ratio =min(hwratio, whratio);double area =rect.area();//框出符合条件的轮廓，舍去: 面积很小的, 面积很大的, 长宽比严重失调的

if (area > minArea && area < img_size*maxAreaRatio && ratio >boxRatio) {int w =rect.width;int h =rect.height;if(rect.width < 20)

rect.width= 20;/*if(rect.width < 6)

rect.width = 2*rect.width;

else

rect.width = rect.width + rect.width/2;*/

if(rect.height < 20)

rect.height= 20;/*if(rect.height < 6)

rect.height = 2*rect.height;

else

rect.height = rect.height + rect.height/2;*/

int w_add = (rect.width - w)/2;int h_add = (rect.height - h)/2;

rect.x= rect.x -h_add;

rect.y= rect.y -w_add;

rectangle(img, rect, Scalar(0,0,255));

}

}

}