Конвертировать данные из cv :: Mat в mxArray - PullRequest
Новая
       7

Конвертировать данные из cv :: Mat в mxArray

7 голосов
/ 12 января 2012

Я хочу отправить Matrix в Matlab, используя "engine.h" в коде c ++. Дело в том, что у меня есть данные внутри 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));

Кто-нибудь знает правильный способ сделать преобразование? Любая помощь будет оценена. Спасибо.

EDIT

Я нашел этот путь: https://stackoverflow.com/a/8848711/744859

Ответы [ 3 ]

10 голосов
/ 24 февраля 2012

Существует библиотека, разработанная Kota Yamaguchi по адресу http://github.com/kyamagu/mexopencv. Пакет содержит класс C ++ (называемый MxArray), который преобразует между собственным типом данных Matlab (mxArray) и типами данных OpenCV.Библиотека напрямую поддерживает API C ++ для OpenCV (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
4 голосов
/ 12 января 2012

В этой теме показано, как преобразовать 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;
}
3 голосов
/ 13 января 2012

Я нашел более простой способ сделать это преобразование после некоторых усилий. Я создаю такую ​​функцию: 

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);
}
...