如何从函数返回多维 Mat 数组

Question

我想返回一个尺寸为 [w][h][c] 的 cv::Mat 对象，其中 w 是宽度，h 是高度，c 是通道或深度。但是编译器给了我一个错误。

这是我的代码：

cv::Mat three_channel_convolution(cv::Mat bgr_image, cv::Mat kernel)
{
   cv::Mat feature_map[3];
   cv::Mat bgr[3];

   cv::split(bgr_image, bgr);

   for (uchar i = 0; i < 3; i++)
   {
      feature_map[i] = one_channel_convolution(bgr[i], kernel);
      std::string name_feature_map = "bgr_feature_map_" + std::to_string(i) + ".jpg";
      cv::imwrite(name_feature_map, feature_map[i]);
   }

return feature_map;
}

在这个函数中，我想为图像 BGR 中的三个通道中的每一个通道使用一个过滤器（内核）进行卷积。所以卷积函数在一个通道中返回运算结果（A single 2D Mat [][]），并将其存储在 feature_map Mat 数组中。如何返回这个 Mat 数组？

然后在我的 main() 函数中，我这样调用函数：

feature_map = three_channel_convolution(image, kernels[0]);

感谢您的帮助！

Answer 1

函数three_channel_convolution的返回值是cv::Mat，但是你返回的是cv::Mat[3]。

如果您想返回一个 3d 矩阵，您需要将函数的签名更改为 std::array<cv::Mat, 3> 之类的东西。请记住，这可能存在一些性能问题。

std::array<cv::Mat, 3> three_channel_convolution(cv::Mat bgr_image, cv::Mat kernel) {
   std::array<cv::Mat, 3> feature_map;
   std::array<cv::Mat, 3> bgr;

   cv::split(bgr_image, bgr);

   for (uchar i = 0; i < 3; i++) {
      feature_map[i] = one_channel_convolution(bgr[i], kernel);
      std::string name_feature_map = "bgr_feature_map_" + std::to_string(i) + ".jpg";
      cv::imwrite(name_feature_map, feature_map[i]);
   }

    return feature_map;
}

附言。请在后续帖子中报告确切的错误消息。