将数据从cv::Mat转换为mxArray

Convert data from cv::Mat to mxArray

本文关键字:转换 mxArray Mat cv 数据      更新时间:2023-10-16

我想使用c++代码中的"engine.h"将矩阵发送到Matlab。事实上,我在cv::Mat中有数据,我需要发送一个mxArray。我试着用这个表达,但它不起作用:

cv::Mat _priorP;
_priorP = Mat::eye(13, 13, CV_32FC1);
mxArray *mat;
mat = mxCreateDoubleMatrix(13, 13, mxREAL);
memcpy(mxGetPr(mat),_priorP.data, 13*13*sizeof(double));

有人知道转换的正确方法吗?任何帮助都是值得的。谢谢

编辑

我是这样发现的:https://stackoverflow.com/a/8848711/744859

Kota Yamaguchi在http://github.com/kyamagu/mexopencv该包包含一个C++类(称为MxArray),用于在Matlab的本地数据类型(MxArray)和OpenCV数据类型之间进行转换。该库直接支持OpenCV的C++API(Open CV 2.0版及更高版本),因此无需进行额外的转换(例如从cvMat到CV::Mat或从IplImage到CV::Mat)。示例用法:

#include "mexopencv.hpp" // include the library
#include "highgui.h"     
void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[]) 
{
    cv::Mat image;
    image = imread("filename"); // read an image from file
    plhs[0] = MxArray(image);   // convert from cv::Mat to mxArray 
}

就是这样。请确保将mex函数与库中的MxArray.cpp文件一起编译;你可以在MATLAB命令行中这样做:

    mex yourmexfile.cpp MxArray.cpp

此线程显示如何将CvMat转换为mxArray。尽管它并不完全是您想要的转换代码,但它非常接近。

这是一个简单的转换,您应该能够调整代码以使用cv::Mat而不是CvMat。如果你做不到,一个快速的破解方法是将你的cv::Mat数据转换为CvMat,然后按原样使用下面的代码(取自我建议的链接):

mxArray* CvMat_to_new_mxArr (const CvMat* mat)
{
  const int TYPE = cvGetElemType (mat);
  // 2-d image
  if (CV_64FC1 == TYPE) {
    return helper_2dcvmat_to_mat<CV_64FC1> (mat);
  }
  else if (CV_32FC1 == TYPE) {
    return helper_2dcvmat_to_mat<CV_32FC1> (mat);
  }
  else if (CV_32SC1 == TYPE) {
    return helper_2dcvmat_to_mat<CV_32SC1> (mat);
  }
  else if (CV_16SC1 == TYPE) {
    return helper_2dcvmat_to_mat<CV_16SC1> (mat);
  }
  else if (CV_16UC1 == TYPE) {
    return helper_2dcvmat_to_mat<CV_16UC1> (mat);
  }
  else if (CV_8UC1 == TYPE) {
    return helper_2dcvmat_to_mat<CV_8UC1> (mat);
  }
  else if (CV_8SC1 == TYPE) {
    return helper_2dcvmat_to_mat<CV_8SC1> (mat);
  }
  //Multi-dimensional arrays not supported, yet.
  /*
  // 3-d image
  else if (CV_64FC3 == TYPE) {
    return helper_rgbimage_to_mat<IPL_DEPTH_64F> (img);
  }
  else if (CV_32FC3 == TYPE) {
    return helper_rgbimage_to_mat<IPL_DEPTH_32F> (img);
  }
  else if (CV_8UC3 == TYPE) {
    return helper_rgbimage_to_mat<IPL_DEPTH_8U> (img);
  }
  */
  // unsupported conversion, return null mxArray
  return mxCreateDoubleMatrix(0,0,mxREAL);    
}

template<int TYPE>
mxArray* helper_2dcvmat_to_mat (const CvMat* mat)
{
  void* pBeg;
  int pitch;
  cvGetRawData(mat, (uchar**)&pBeg,&pitch);
  CvSize size = cvGetSize (mat);
  const mxClassID cid = cvm_traits<TYPE>::CID;
  mxArray* pArrOut =
  mxCreateNumericMatrix(size.height,size.width,cid,mxREAL);
  void* pBegOut = mxGetData(pArrOut);
  typedef mc_traits<cid>::CT T;
  pix_iterator_2d<T,eRowWise> it_src1(static_cast<T*>(pBeg),
  size.width,size.height,pitch);
  pix_iterator_2d<T,eRowWise> it_src2(static_cast<T*>(pBeg),
  size.width,size.height,pitch);
  it_src2.end ();
  pix_iterator_2d<T,eColWise> it_dest(static_cast<T*>(pBegOut),
  size.width,size.height);
  std::copy (it_src1,it_src2,it_dest);
  return pArrOut;
}

经过一番努力,我找到了一种更简单的方法来进行转换。我所做的是创建这样一个函数:

void arithmetic::cvLoadMatrixToMatlab( Engine *ep, const Mat& m, string name)
{   
   int rows=m.rows;
   int cols=m.cols;
   //Mat data is float, and mxArray uses double, so we need to convert.   
   mxArray *T=mxCreateDoubleMatrix(cols, rows, mxREAL);
   double *buffer=(double*)mxGetPr(T);
   for(int i=0; i<rows; i++){
       for(int j=0; j<cols; j++){
           buffer[i*(cols)+j]= (double)m.at<float>(i, j);
       }
   }
   //memcpy((char*)mxGetPr(T), (char*)m.data, rows*cols*sizeof(double));
   engPutVariable(ep, name.c_str(), T);
   name=name+"="+name+"'";                    // Column major to row major (mat=mat')
   engEvalString(ep, name.c_str());
   mxDestroyArray(T);
}