Grabcut的算法实现的流程:
**********参数说明****************
GrabCut函数使用
在OpenCV的源码目录的samples的文件夹下,有grabCut的使用例程,请参考:
opencv\samples\cpp\grabcut.cpp。
而grabCut函数的API说明如下:
void cv::grabCut( InputArray _img, InputOutputArray _mask, Rect rect,
InputOutputArray _bgdModel, InputOutputArray _fgdModel,
int iterCount, int mode )
/*
****参数说明:
img——待分割的源图像,必须是8位3通道(CV_8UC3)图像,在处理的过程中不会被修改;
mask——掩码图像,如果使用掩码进行初始化,那么mask保存初始化掩码信息;在执行分割的时候,也可以将用户交互所设定的前景与背景保存到mask中,然后再传入grabCut函数;在处理结束之后,mask中会保存结果。mask只能取以下四种值:
GCD_BGD(=0),背景;
GCD_FGD(=1),前景;
GCD_PR_BGD(=2),可能的背景;
GCD_PR_FGD(=3),可能的前景。
如果没有手工标记GCD_BGD或者GCD_FGD,那么结果只会有GCD_PR_BGD或GCD_PR_FGD;
rect——用于限定需要进行分割的图像范围,只有该矩形窗口内的图像部分才被处理;
bgdModel——背景模型,如果为null,函数内部会自动创建一个bgdModel;bgdModel必须是单通道浮点型(CV_32FC1)图像,且行数只能为1,列数只能为13x5;
fgdModel——前景模型,如果为null,函数内部会自动创建一个fgdModel;fgdModel必须是单通道浮点型(CV_32FC1)图像,且行数只能为1,列数只能为13x5;
iterCount——迭代次数,必须大于0;
mode——用于指示grabCut函数进行什么操作,可选的值有:
GC_INIT_WITH_RECT(=0),用矩形窗初始化GrabCut;
GC_INIT_WITH_MASK(=1),用掩码图像初始化GrabCut;
GC_EVAL(=2),执行分割。
*/
下面放上程序:#include<opencv2\highgui\highgui.hpp>
#include<opencv2\imgproc\imgproc.hpp>
#include<iostream>
using namespace cv;
Point Point1, Point2;
Rect rect;
Mat srcImage, roiImage;
void on_MouseHandle(int event, int x, int y, int flags, void* param);//event鼠标事件代号,x,y鼠标坐标,flags拖拽和键盘操作的代号
int main()
{
srcImage = imread("flower.jpg");
if (!srcImage.data)
{
printf("could not load image...\n");
return -1;
}
namedWindow("srcImage", CV_WINDOW_AUTOSIZE);
imshow("srcImage", srcImage);
//定义前景与输出图像
Mat foreground(srcImage.size(), CV_8UC3, Scalar(255, 255, 255));//定义一张三通道的前景图
Mat result(srcImage.size(), CV_8UC1);//定义输出图像
//鼠标响应函数、
setMouseCallback("srcImage", on_MouseHandle, 0);
//Grabcut分割前景与背景
Mat fgMat, bgMat;//定义前景与背景
//定义迭代次数
int i = 3;
//选择区域有效,按下Esc键退出
while (true)
{
char c = (char)waitKey(0);
if (c == 'n')
{
//实现图割操作
grabCut(srcImage, result, rect, bgMat, fgMat, i, GC_INIT_WITH_RECT);
//图像匹配
compare(result, GC_PR_FGD, result, CMP_EQ);//比较result的值可能是前景才输出到result中
//生成前景图像
srcImage.copyTo(foreground, result);//将reuslt的结果覆盖到foreground中
imshow("foreground", foreground);
}
if ((int)c == 27)
{
break;
}
}
waitKey(0);
return 0;
}
void on_MouseHandle(int event, int x, int y, int flags, void* param)
{
if (event == CV_EVENT_LBUTTONDOWN)//如果鼠标的左键按下的话,读取初始坐标
{
Point1 = Point(x, y);//鼠标的初始坐标
}
else if (event == EVENT_MOUSEMOVE && (flags & CV_EVENT_FLAG_LBUTTON))//左键按下的时候,鼠标移动,则在图像上画矩形
{
Mat img1 = srcImage.clone();//克隆一张新的图
Point2 = Point(x, y);//鼠标的当前的坐标
//在图形上绘制矩形区域
rectangle(img1, Point1, Point2, Scalar(0, 0, 255), 2, 8, 0);
imshow("image", img1);
}
else if (event == CV_EVENT_LBUTTONUP)//鼠标的左键松开的时候
{
Point2 = Point(x, y);
rect = Rect(Point1.x, Point1.y, Point2.x - Point1.x, Point2.y - Point1.y);
//roiImage = srcImage(rect);
//imshow("roiImage", roiImage);
}
}
原图: