将图像输出回Matlab Mex

Question

我正在尝试将图像从我的mex文件输出回我的matlab文件，但是当我在matlab中打开它时，它是不正确的。

带有mex文件的输出图像是正确的

我试图切换mwSize的方向以及在new_img.at<int>(j, i)交换i和j ;

Mat image = imread(mxArrayToString(prhs[0]));
Mat new_img(H,W, image.type(), Scalar(0));
// some operations on new_img
imshow( "gmm image", image ); //shows the original image
imshow( "gmm1 image", new_img ); //shows the output image
waitKey( 200 );  //both images are the same size as desired

mwSize nd = 2;
mwSize dims[] = {W, H};

plhs[0] = mxCreateNumericArray(nd, dims, mxUINT8_CLASS, mxREAL);
if(plhs == NULL) {
  mexErrMsgTxt("Could not create mxArray.\n");     
}     
char* outMat = (char*) mxGetData( plhs[0]);

for (int i= 0; i < H; i++)
{
  for (int j = 0; j < W; j++)
  {       
    outMat[i +j*image.rows] = new_img.at<int>(j, i);       
  }
}

这在垫子文件中

gmmMask = GmmMex2(imgName,rect);
imshow(gmmMask); % not the same as the output image. somewhat resembles it, but not correct.

Answer 1

因为您已经暗示这是彩色图像 ，所以这意味着您要考虑矩阵的三个部分。 您的代码仅考虑一片。 首先，您需要确保声明正确的图像尺寸。 在MATLAB中，第一维始终是行数，而第二维是列数。 现在，您还必须在此之上添加通道数。 我假设这是RGB图像，所以有三个通道。

因此，将您的dims更改为：

mwSize nd = 3;
mwSize dims[] = {H, W, nd};

将nd更改为3很重要，因为这将允许您创建3D矩阵。 您只有2D矩阵。 接下来，确保要访问cv::Mat对象中正确位置的图像像素。 在嵌套的for循环对中访问图像像素的方式采用行为主的方式（先在列上迭代，然后在行上迭代）。 这样，您需要在i访问行和j访问列时交换i和j 。 您还需要访问彩色图像的通道，因此需要另一个for循环来进行补偿。 对于灰度情况，尽管如此，您已经适当地补偿了MATLAB MEX矩阵的列主内存配置。 这已得到验证，因为j访问列，并且您需要按行数跳过才能访问下一列。 但是，要适应彩色图像，还必须按image.rows*image.cols跳过以转到下一层像素。

因此，您的for循环现在应为：

for (int k = 0; k < nd; k++) {
    for (int i = 0; i < H; i++) { 
        for (int j = 0; j < W; j++) {
            outMat[k*image.rows*image.cols + i + j*image.rows] = new_img.at<uchar>(i, j, k);
        }
    }
}

请注意，像素容器很可能是8位无符号字符，因此必须将模板更改为uchar not int 。 这也可以解释您的程序为什么崩溃的原因。