OpenCV - Mat的零填充

时间:2015-05-26 22:28:58

标签: c++ opencv padding

假设我有一个Mat变量small_image,其尺寸为98x158x32(类型为float)。现在我想零填充此图像(即向图像添加零边框)。我想在图像的上方和下方添加7个零,并在图像的左侧和右侧添加12个零。第一个想法是使用cv copyMakeBorder(参见copyMakeBorder doc),这似乎是完美的:

    int old_size[3];
    old_size[0] = 98;
    old_size[1] = 158;
    old_size[2] = 32;

    int pad_size[3];
    pad_size[0] = old_size[0] + 2 * 7;
    pad_size[1] = old_size[1] + 2 * 12;
    pad_size[2] = old_size[2];

    cv::Mat image_padded(3, pad_size, CV_32FC1, cv::Scalar(0)); //Initialize the larger Mat to 0

    copyMakeBorder(small_image,image_padded,7,7,12,12,BORDER_CONSTANT,Scalar(0));

但是,此代码会出现memcopy错误。有没有人在这看到问题?

post中描述的替代方法也不起作用:

cv::Rect roi( cv::Point( 12, 7 ), small_image.size() );
small_image.copyTo( image_padded( roi ) );

它声称"断言失败(m.dims< = 2)",而两个Mat变量都是3D矩阵。

非常感谢任何帮助实现零填充!

2 个答案:

答案 0 :(得分:2)

您描述的方法都不起作用,因为它们用于2D矩阵。你有一个3D Mat,要求你在所有三个维度中指定范围。您可以使用()运算符,该运算符使用与copyTo结合的范围数组,如下例所示。

#include <opencv2/imgproc/imgproc.hpp>
#include <iostream>

int main()
{
  int old_size[3];
  old_size[0] = 5;
  old_size[1] = 5;
  old_size[2] = 2;

  int pad_size[3];
  pad_size[0] = old_size[0] + 2 * 1;
  pad_size[1] = old_size[1] + 2 * 2;
  pad_size[2] = old_size[2];

  cv::Mat small_image(3, old_size, CV_32FC1, cv::Scalar(1));
  cv::Mat image_padded(3, pad_size, CV_32FC1, cv::Scalar(0));

  cv::Range ranges[3];
  ranges[0] = cv::Range(1, old_size[0]+1);
  ranges[1] = cv::Range(2, old_size[1]+2);
  ranges[2] = cv::Range(0, old_size[2]);

  small_image.copyTo(image_padded(ranges));

  for (int i = 0; i < pad_size[0]; i++)
    for (int j = 0; j < pad_size[1]; j++)
      for (int k = 0; k < pad_size[2]; k++)
        std::cout << image_padded.at<float>(i, j, k) << ", ";
}

将给出:

0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1,
1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1,
1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 
0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0

答案 1 :(得分:1)

我们还可以使用rowRangecolRange方法的组合将我们感兴趣的区域设置为3维填充图像(全零)的子部分。这可以很容易地将未打印的图像复制到其上。

cv::Mat image_padded(3, pad_size, CV_32FC1, cv::Scalar(0));
int xPad = 12;
int yPad = 7;
Mat area = image_padded.rowRange(xPad, xPad + 98).colRange(yPad, yPad + 158);
small_image.copyTo( area );