Matlab调用VIBE算法

Matlab调用VIBE算法

从参考[1]中下载源码，VIBE的算法实现由vibe-background-sequential.c和vibe-background-sequential.h构成，为了mex编译方便将vibe-background-sequential.c文件的后缀改为cpp。

mex的接口文件为VIBE_.cpp：

// VIBE_.cpp

// ------------------------------------------------------------------ -

// Reference: your-main-file-sequential.c and mexopencv-master/BackgroundSubtractorMOG2_.cpp

// Authors : LSS

// Date : 26 / 04 / 2017

// Last modified : 28 / 04 / 2017

// ------------------------------------------------------------------ -

#include "mex.h"

#include "matrix.h" // for the mxArray

#include

#include

#include

#include

#include "vibe-background-sequential.h"

#include "vibe-background-sequential.cpp"

using namespace std;

class VIBE

{

private:

vibeModel_Sequential_t *model;

bool bfirstFrame;

public:

VIBE()

{

model = (vibeModel_Sequential_t*)libvibeModel_Sequential_New();

bfirstFrame = true;

};

VIBE(const VIBE & vibe)

{

model = (vibeModel_Sequential_t*)libvibeModel_Sequential_New();

/* Default parameters values. */

model->numberOfSamples = vibe.model->numberOfSamples;

model->matchingThreshold = vibe.model->matchingThreshold;

model->matchingNumber = vibe.model->matchingNumber;

model->updateFactor = vibe.model->updateFactor;

/* Storage for the history. */

model->historyImage = vibe.model->historyImage;

model->historyBuffer = vibe.model->historyBuffer;

model->lastHistoryImageSwapped = vibe.model->lastHistoryImageSwapped;

/* Buffers with random values. */

model->jump = vibe.model->jump;

model->neighbor = vibe.model->neighbor;

model->position = vibe.model->position;

bfirstFrame = true;

};

~VIBE()

{

libvibeModel_Sequential_Free(model);

};

VIBE & operator=(const VIBE & vibe)

{

if (this == &vibe)

return *this;

/* Default parameters values. */

model->numberOfSamples = vibe.model->numberOfSamples;

model->matchingThreshold = vibe.model->matchingThreshold;

model->matchingNumber = vibe.model->matchingNumber;

model->updateFactor = vibe.model->updateFactor;

/* Storage for the history. */

model->historyImage = vibe.model->historyImage;

model->historyBuffer = vibe.model->historyBuffer;

model->lastHistoryImageSwapped = vibe.model->lastHistoryImageSwapped;

/* Buffers with random values. */

model->jump = vibe.model->jump;

model->neighbor = vibe.model->neighbor;

model->position = vibe.model->position;

bfirstFrame = true;

return *this;

};

void GetfrImageC3R(uint8_t * image_data, uint8_t * frimg, int width, int height) // image_data is stored as RGBRGBRGB... or BGRBGRBGR... Three consecutives bytes per pixel thus.

{

if (bfirstFrame)

{

libvibeModel_Sequential_AllocInit_8u_C3R(model, image_data, width, height);

bfirstFrame = false;

}

else

{

libvibeModel_Sequential_Segmentation_8u_C3R(model, image_data, frimg);

libvibeModel_Sequential_Update_8u_C3R(model, image_data, frimg);

}

}

};

/// Last object id to allocate

int last_id = 0;

/// Object container

map obj_;

void mexFunction(int nlhs, mxArray *plhs[],

int nrhs, const mxArray *prhs[])

{

// Checking number of arguments

if (nrhs<2 || nlhs>1)

mexErrMsgIdAndTxt("VIBE : error", "Wrong number of arguments");

int id = 0;

string method;

if (nrhs > 1 && mxIsNumeric(prhs[0]) && mxIsChar(prhs[1]))

{

id = mxGetScalar(prhs[0]);

method = mxArrayToString(prhs[1]);

}

else

mexErrMsgIdAndTxt("VIBE : error", "Invalid arguments");

if (method == "new")

{

obj_[last_id] = VIBE(); // 调用了两次默认构造函数，和一次赋值函数

plhs[0] = mxCreateDoubleScalar(last_id++);

return;

}

VIBE& obj = obj_[id];

if (method == "delete")

{

if (nrhs != 2 || nlhs != 0)

mexErrMsgIdAndTxt("VIBE : error", "Output not assigned");

obj_.erase(id);

}

else if (method == "GetfrImageC3R")

{

if (nrhs != 3 || nlhs != 1)

mexErrMsgIdAndTxt("VIBE : error", "Wrong number of arguments");

if (!mxIsClass(prhs[2], "uint8"))

mexErrMsgIdAndTxt("VIBE : error", "Only image arrays of the uint8 class are allowed");

uint8_t* imMatlab = (uint8_t*)mxGetPr(prhs[2]);

int nrDims = (int)mxGetNumberOfDimensions(prhs[2]);

if (nrDims != 3)

{

mexErrMsgIdAndTxt("VIBE : error", "The input image should be RGB");

return;

}

int width = (mxGetDimensions(prhs[2]))[0];

int height = (mxGetDimensions(prhs[2]))[1];

uint8_t *image_data = NULL;

image_data = (uint8_t*)mxMalloc((3 * width) * height);

for (int i = 0; i < width; i++) // image_data is stored should as RGBRGBRGB

{

for (int j = 0; j < height; j++)

{

image_data[j * 3 * width + i*3] = imMatlab[j*width + i];

image_data[j * 3 * width + i*3 + 1] = imMatlab[width*height + j*width + i];

image_data[j * 3 * width + i*3 + 2] = imMatlab[2*width*height + j*width + i];

}

}

plhs[0] = mxCreateNumericMatrix(width, height, mxUINT8_CLASS, mxREAL);

uint8_t * frimg = (uint8_t *) mxGetPr(plhs[0]);

memset(frimg, 0, sizeof(uint8_t)*width*height);

obj.GetfrImageC3R(image_data, frimg, width, height);

mxFree(image_data);

}else

mexErrMsgIdAndTxt("VIBE : error", "Unrecognized operation");

return;

}

并在Matlab环境下定义VIBE类(VIBE.m)：

classdef VIBE < handle

% VIBE.m

% -------------------------------------------------------------------

% Reference: BackgroundSubtractorMOG2.m

% Authors: LSS

% Date: 27/04/2017

% Last modified: 28/04/2017

% -------------------------------------------------------------------

properties (SetAccess = private)

id % Object ID

end

methods

function this = VIBE()

this.id = VIBE_(0, 'new');

end

function delete(this)

%DELETE Destructor

%

VIBE_(this.id, 'delete');

end

function fgmask = GetfrImageC3R(this, im)

%

fgmask = VIBE_(this.id, 'GetfrImageC3R', im);

end

end

end

mex的编译代码为：

cmd = ['mex -largeArrayDims VIBE_.cpp'];

eval(cmd)

调用VIBE的示例代码为：

% demo_camera.m

% -------------------------------------------------------------------

% Reference:

% Authors: LSS

% Date: 27/04/2017

% Last modified: 28/04/2017

% -------------------------------------------------------------------

videoObj = VideoReader('E:\code\vcproject\FireFogVS2013-2017-24-camera\FireFogVS2013\out3.avi');

numFrames = get(videoObj, 'NumberOfFrames');

FRAME_HEIGHT = 480;

FRAME_WIDTH = 640;

vibe = VIBE();

for ii = 1:numFrames-1,

im = read(videoObj, ii);

im = double(imresize(im, [FRAME_HEIGHT, FRAME_WIDTH]));

frimg = vibe.GetfrImageC3R(uint8(im));

figure(50), imshow(frimg)

end

实验结果图像为：

上面的所有代码可以在链接[2]中下载。

参考：

【1】http://www.telecom.ulg.ac.be/research/vibe/doc/

【2】VIBE-MEX--   http://download.csdn.net/detail/lsxpu/9828310

Matlab调用VIBE算法相关教程