将OpenCV Mat发送到MATLAB工作区而不复制数据的方法?

时间:2014-12-16 21:55:27

标签: c++ matlab opencv mex

当我编写使用OpenCV函数的MEX文件时,很容易将数据 MATLAB传递到MEX环境而无需复制数据。有没有办法以相同的方式将数据返回到 MATLAB? (也就是说,不复制数据而不会导致MATLAB崩溃......)

一个简单的例子:

#include "mex.h"
#include "/opencv2/core.hpp"
void mexFunction(int nlhs, mxArray *plhs[], int nrhs,const mxArray *prhs[])
{

    Rows=mxGetM(prhs[0]);
    Cols=mxGetN(prhs[0]);
    Mat InMat(Cols,Rows,CV_64FC1,mxGetPr(prhs[0]));//Matlab passes data column-wise 
                                                   // no need to copy data - SUPER!

    InMat=InMat.t();//transpose so the matrix is identical to MATLAB's one

    //Make some openCV operations on InMat to get OutMat...


    //Way of preventing the following code??

    plhs[0]=mxCreateDoubleMatrix(OutMat.rows,OutMat.cols,mxREAL);
    double *pOut=mxGetPr(plhs[0]);

    for (int i(0);i<OutMat.rows;i++)
      for (int j(0);j<OutMat.cols;j++)
         pOut[i+j*OutMat.rows]=OutMat.at<double>(i,j);

}

1 个答案:

答案 0 :(得分:3)

通常我就像那样做输入和输出,附加一个指针来处理输入并循环输出上的元素。但是,我认为输出可以以与输入类似的方式完成,但不是没有某种副本。避免复制的唯一方法是使用来自Mat的指针创建输出mxArray并在其中进行操作。当然,这并不总是可行的。但是你可以优雅地了解如何复制数据。

您可以利用将缓冲区附加到您使用的cv::Mat的同一技巧(我也是!)从MATLAB中输入数据,但也可以将其输出。导出数据的技巧的转折是恰到好处地使用copyTo,以便它将使用现有缓冲区,即mxArrayplhs[i]的缓冲区。

从这样的输入开始:

double *img = mxGetPr(prhs[0]);
cv::Mat src = cv::Mat(ncols, nrows, CV_64FC1, img).t(); // nrows <-> ncols, transpose

您执行某些操作,例如调整大小:

cv::Mat dst;
cv::resize(src, dst, cv::Size(0, 0), 0.5, 0.5, cv::INTER_CUBIC);

要使dst进入MATLAB:首先转置输出(为了将数据重新排序为大写顺序)然后创建输出{{ 1}}使用指向cv::Mat plhs[0]的指针,终于调用mxArray以使用转置数据填充包装copyTo:< / p>

Mat

为了避免重新分配dst = dst.t(); // first! cv::Mat outMatWrap(dst.rows, dst.cols, dst.type(), pOut); // dst.type() or CV_* dst.copyTo(outMatWrap); // no realloc if dims and type match 而对copyTo的以下调用获取完全相同的维度和数据类型非常重要。

请注意,当outMatWrap被销毁时,outMatWrap缓冲区将不会被释放,因为引用计数为0(data不会解除分配Mat::release())。


可能的模板(绝不是防弹!)

.data

这对于通道&gt; 1应该是好的,只要MATLAB阵列的大小也是像素顺序(例如3xMxN)。然后根据需要使用template <typename T> void cvToMATLAB(cv::Mat mat, T *p) { CV_Assert(mat.elemSize1() == sizeof(T)); mat = mat.t(); cv::Mat outMatWrap(mat.rows, mat.cols, mat.type(), p); mat.copyTo(outMatWrap); }


关于permute

的说明

copyTo将重新分配目标缓冲区的条件是维度或数据类型不匹配:

opencv2 \ core \ mat.hpp 第347行(版本2.4.10),附有我的评论:

copyTo

因此,只需确保您获得正确的大小和数据类型,数据最终将在inline void Mat::create(int _rows, int _cols, int _type) { _type &= TYPE_MASK; if( dims <= 2 && rows == _rows && cols == _cols && type() == _type && data ) return; // HIT THIS TO USE EXISTING BUFFER! int sz[] = {_rows, _cols}; create(2, sz, _type); // realloc! } 缓冲区中而不是其他位置。如果你做得对,mxArray将使用你指定的缓冲区,在每一行调用copyTo