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 <map>#include <cstdint>#include <vector>#include <string>#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 allocateint last_id = 0;/// Object containermap<int, VIBE> obj_;void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[]){// Checking number of argumentsif (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]);}elsemexErrMsgIdAndTxt("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);}elsemexErrMsgIdAndTxt("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