将图像输出回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);       
  }
}

这是在 mat 文件中

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 矩阵。你只有一个二维矩阵。接下来，确保您正在访问cv::Mat 对象中正确位置的图像像素。您在嵌套的一对for 循环中访问图像像素的方式采用row-major 方式（先遍历列，然后遍历行）。因此，您需要交换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 不是 int。这也可以解释您的程序崩溃的原因。