2

我有一个要从其他人的 C# 应用程序调用的 C++ 函数。作为输入,我的函数得到一个有符号短整数数组、它所代表的图像的尺寸以及为返回数据分配的内存,即另一个有符号短整数数组。这将代表我的函数的标题:

my_function (short* input, int height, int width, short* output)

在我的函数中,我从 cv::Mat 创建了一个input,如下所示:

cv::Mat mat_in = cv::Mat (height, width, CV_16S, input);

然后mat_in将其转换为CV_32FOpenCV 的cv::bilateralFilter. 在它返回 cv::Mat mat_out 之后,我将数据转换回CV_16SbilateralFilter只接受CV_8UCV_32F)。现在我需要将它转换cv::Mat mat_out回一个短整数数组,以便它可以返回给调用函数。这是我的代码:

my_function (short* input, int height, int width, short* output)
{
    Mat mat_in_16S = Mat (height, width, CV_16S, input);

    Mat mat_in_32F = Mat (height, width, CV_32F);
    Mat mat_out_CV_32F = Mat (height, width, CV_32F);

    mat_in_16S.convertTo (mat_in_32F, CV_32F);

    bilateralFilter (mat_in_32F, mat_out_32F, 5, 160, 2);
    Mat mat_out_16S = Mat (mat_in_16S.size(), mat_in_16S.type());
    mat_out_32F.convertTo (mat_out_16S, CV_16S);

    return 0;
}

显然,在最后的某个地方,我需要将数据mat_out_16S放入output. 我的第一次尝试是返回参考:

output = &mat_out_16S.at<short>(0,0);

但是我当然意识到这是一个愚蠢的想法,mat_out_16S因为一旦函数返回就超出范围,output指向空虚。目前我最好的尝试如下(来自这个问题):

memcpy ((short*)output, (short*)mat_out_16S.data, height*width*sizeof(short));

现在我想知道,有没有更好的方法?复制所有这些数据感觉有点低效,但我看不出我还能做什么。不幸的是,我无法返回 cv::Mat。如果没有更好的方法,我目前的memcpy方法至少安全吗?我的数据都是 2 字节有符号短整数,所以我认为填充不应该有问题,但我不想遇到任何不愉快的意外。

4

1 回答 1

1

您可以将此构造函数用于您的mat_out_16S

Mat::Mat(Size size, int type, void* data, size_t step=AUTO_STEP)

所以你的功能将是:

my_function (short* input, int height, int width, short* output)
{
    Mat mat_in_16S = Mat (height, width, CV_16S, input);

    Mat mat_in_32F = Mat (height, width, CV_32F);
    Mat mat_out_CV_32F = Mat (height, width, CV_32F);

    mat_in_16S.convertTo (mat_in_32F, CV_32F);

    bilateralFilter (mat_in_32F, mat_out_32F, 5, 160, 2);
    Mat mat_out_16S = Mat (mat_in_16S.size(), mat_in_16S.type(), output);
    mat_out_32F.convertTo (mat_out_16S, CV_16S);

    return 0;
}
于 2013-05-02T13:53:14.970 回答