2012-01-12 20 views
7

Deseo enviar una matriz a Matlab usando "engine.h" en código C++. El hecho es que tengo los datos dentro de un cv :: Mat, y necesito enviar un mxArray. Intenté usar esta expresión pero no funciona:Convertir datos de cv :: Mat en 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)); 

¿Alguien sabe la forma correcta de hacer la conversión? Cualquier ayuda sería apreciada. Gracias.

EDITAR

He encontrado de esta manera: https://stackoverflow.com/a/8848711/744859

Respuesta

4

This thread muestra cómo convertir un CvMat a mxArray. Aunque no es exactamente el código de conversión que está buscando, es bastante cercano.

Esta es una conversión simple y debería poder ajustar el código para trabajar con cv::Mat en lugar de CvMat. Si no puede, un corte rápido es convertir sus datos cv::Mat a CvMat y luego usar el siguiente código como está (tomada desde el enlace sugerí):

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; 
} 
+0

Gracias. Entonces, siempre es complicado usar engine.h con openCV, ¿no? Esperaba algo más directo, pero si no hay una solución más simple, probaré esta. –

+1

Una pregunta. ¿Está considerando que OpenCV almacene datos por fila y matlab por columna? –

+1

Debo insistir en que abra el hilo que sugerí y eche un vistazo al código fuente ** completo ** disponible allí. Según los otros usuarios, ese código funciona. Pero si no lo hace, al menos es un comienzo para ti. – karlphillip

3

he encontrado un método más simple de hacer esto conversión después de un poco de esfuerzo. Lo que hago es crear una función como esta:

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

Hay una biblioteca desarrollada por Kota Yamaguchi en http://github.com/kyamagu/mexopencv El paquete contiene una clase de C++ (llamado MxArray) que convierte entre el tipo de datos nativo de Matlab (mxArray) y Tipos de datos OpenCV. La biblioteca admite directamente la API de C++ para OpenCV (Open CV versión 2.0 y superior), por lo que no es necesario realizar conversiones adicionales (por ejemplo, de cvMat a cv :: Mat o de IplImage a cv :: Mat). Uso de la muestra:

#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 
} 

Eso es todo. Asegúrese de compilar su función mex junto con archivo MxArray.cpp de la biblioteca; puede hacerlo en la línea de comando de MATLAB:

mex yourmexfile.cpp MxArray.cpp 
+0

El código fuente es suficiente para mí, solo quiero hacer mi propia función. Muchas gracias, muy útil de hecho ... –

+1

@Jav_Rock ¡Me alegra oír! MxArray.cpp fue muy útil para mí también. Considere votar esta respuesta para que otros puedan ver esta solución. – Alexey

+3

¡sí, ya lo hice! –

Cuestiones relacionadas