Извлечение прозрачного фона изображения с opencv

Question

У меня есть маска, вычисленная в grab_cut (которая вычисляет передний план). Я хочу извлечь только фон, оставляя прозрачный передний план. Мне удается сделать это, используя следующий код, чтобы извлечь передний план (фоновый прозрачный). Как можно сделать наоборот?

int border = 20; 
int border2 = border + border; 
cv::Rect rectangle(border,border,image.cols-border2,image.rows-border2); 

cv::Mat result; // segmentation result (4 possible values) 
cv::Mat bgModel,fgModel;/
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 
cv::compare(result,cv::GC_PR_FGD,result,cv::CMP_EQ); 
cv::Mat foreground(image.size(),CV_8UC3,cv::Scalar(255,255,255)); 
image.copyTo(foreground,result); // bg pixels not copied 
cv::rectangle(image, rectangle, cv::Scalar(255,255,255),1); 
cv::imwrite(argv[2], foreground); 
cv::imwrite(argv[3], image); 

Mat dst;//(src.rows,src.cols,CV_8UC4); 
Mat tmp,alpha; 

cvtColor(foreground,tmp,CV_BGR2GRAY); 
threshold(tmp,alpha,100,255,THRESH_BINARY); 
Mat rgb[3]; 
split(foreground,rgb); 
Mat rgba[4]={rgb[0],rgb[1],rgb[2],alpha}; 
merge(rgba,4,dst); 
imwrite("dst.png",dst); 

В принципе я думаю, что я в должен изменить эти строки:

cv::Mat foreground(image.size(),CV_8UC3,cv::Scalar(255,255,255)); 
image.copyTo(foreground,result); // bg pixels not copied 

Как можно выбрать остальную часть изображения противоположности результата?

Answer 1

Просто инвертировать маску, как в:

cv::Mat background(image.size(),CV_8UC3,cv::Scalar(255,255,255)); 
image.copyTo(background, ~result); // fg pixels not copied