Question

我有一张图片（.jpg图片），我想从原始图片中提取背景。我已经google了很多，但只找到了提取前景图像的教程。

我从另一个stackoverflow question获取了代码。代码对我来说很好，我已成功提取前景（根据我的要求）。现在我想从原始图像中完全删除此前景。我希望它是这样的： -

背景=原始图像 - 前景

空白区域可以填充黑色或白色。我怎样才能做到这一点？

我尝试过使用这种技术： -

Mat background = image2 - foreground;

但它提供了完整的黑色图像。

代码： -

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

using namespace cv;
using namespace std;

int main( )
{
// Open another image
Mat image;
image= cv::imread("images/abc.jpg");

Mat image2 = image.clone();

// define bounding rectangle
cv::Rect rectangle(40,90,image.cols-80,image.rows-170);

cv::Mat result; // segmentation result (4 possible values)
cv::Mat bgModel,fgModel; // the models (internally used)

// GrabCut segmentation
cv::grabCut(image,    // input image
            result,   // segmentation result
            rectangle,// rectangle containing foreground
            bgModel,fgModel, // models
            1,        // number of iterations
            cv::GC_INIT_WITH_RECT); // use rectangle
cout << "oks pa dito" <<endl;
// Get the pixels marked as likely foreground
cv::compare(result,cv::GC_PR_FGD,result,cv::CMP_EQ);
// Generate output image
cv::Mat foreground(image.size(),CV_8UC3,cv::Scalar(255,255,255));
//cv::Mat background(image.size(),CV_8UC3,cv::Scalar(255,255,255));
image.copyTo(foreground,result); // bg pixels not copied

// draw rectangle on original image
cv::rectangle(image, rectangle, cv::Scalar(255,255,255),1);

imwrite("img_1.jpg",image);

imwrite("Foreground.jpg",foreground);
Mat background = image2 - foreground;
imwrite("Background.jpg",background);

return 0;
}

注意：我是一名opencv初学者，现在对它并不了解。如果你能发布完整的代码（根据我的要求）或者只是发布代码行并告诉我这些代码行放在哪里，我将非常感谢你。感谢。

P.S。这是我在StackOverflow.com上的第二个问题。道歉......如果不遵守任何惯例。

Answer 1

不是复制所有前景像素，而是复制所有不是前景的像素。您可以使用~来取消掩码：

image.copyTo(background,~result);

Answer 2

如果你//Get the pixels marked as likely background：

怎么办？

// Get the pixels marked as likely background
cv::compare(result,cv::GC_PR_BGD,result,cv::CMP_EQ);

修改：上面的代码缺少GC_BGD像素。尽管给出了更有效的答案，但让我们完成我们的开始：

// Get the pixels marked as background
cv::compare(result,cv::GC_BGD,result_a,cv::CMP_EQ);
// Get the pixels marked as likely background
cv::compare(result,cv::GC_PR_BGD,result_b,cv::CMP_EQ);
// Final results
result=result_a+result_b;

Answer 3

只是一个小建议，@William's 答案可以写得更简洁：

result = result & 1;

以获取二进制掩码。

Answer 4

也许另一个例子有帮助，其中我假设图像的中间部分肯定是前景。所以试试这个链接。 enter link description here

使用GrabCut提取背景图像

4 个答案: