【OpenCV学习笔记】三十三、GrabCut & FloodFill图像分割

GrabCut & FloodFill图像分割

1.grabCut图像分割算法——grabCut()

2.漫水填充算法——floodFill()

先上ppt：

代码：1.利用Rect做GrabCut分割

///利用Rect做GrabCut分割
#include "opencv2/opencv.hpp"
using namespace cv;
Mat srcImg = imread("bird.jpg", CV_LOAD_IMAGE_COLOR);
Mat tempImg = srcImg.clone();
Mat copyImg = srcImg.clone();
Mat result;//定义输出图像
Rect rect;//定义Rect区域，包含分割对象的矩形ROI, 矩形外部的像素为背景, 矩形内部的像素为前景，当参数mode=GC_INIT_WITH_RECT使用
Mat bgdModel, fgdModel;//背景模型(内部使用)、前景模型(内部使用)
bool flag = false;//记录鼠标左键是否抬起的flag
Point start;//记录鼠标左键按下时的点
void onMouse(int event,int x, int y, int flags, void* param)
{
	switch (event)
	{
	case CV_EVENT_LBUTTONDOWN://鼠标左键按下
		flag = true;//flag置为true
		start = Point(x,y);//记录左键按下时的坐标点
		break;
	case CV_EVENT_MOUSEMOVE://鼠标移动
		if (flag)//如果flag为true
		{
			tempImg = copyImg.clone();//将tempImg置为原图的拷贝copyImg
			rectangle(tempImg, start, Point(x, y), Scalar(0, 255, 0), 1, 8);//画出矩形
			rect = Rect(start, Point(x, y));//定义Rect区域
		}
		break;
	case CV_EVENT_LBUTTONUP://鼠标左键抬起
		flag = false;//flag置为false
		break;
	default:
		break;
	}
	imshow("GrabCut", tempImg);//显示绘制上矩形的tempImg
}
int main()
{
	namedWindow("GrabCut",CV_WINDOW_AUTOSIZE);//新建窗口
	setMouseCallback("GrabCut",onMouse,0);//设置鼠标回调函数，依附于"GrabCut"窗口
	imshow("GrabCut", tempImg);//使得一开始就显示"GrabCut"窗口
	while (1)//无限循环，等待按键
	{
		if (waitKey(0) == 32)//如果按下空格键
		{
			//grabCut分割
			//注意：结果图像result中只有四种可能的取值：GC_BGD、GC_FGD、GC_PR_BGD及GC_PR_FGD
			grabCut(srcImg, result, rect, bgdModel, fgdModel, 1, GC_INIT_WITH_RECT);
			//CMP_EQ参数，将result中逐个像素与GC_PR_FGD比较，寻找出相同的像素，结果存进result中
			compare(result, GC_PR_FGD, result, CMP_EQ);
			imshow("result", result);
		}
	}
	waitKey(0);
	return 0;
}

运行结果：

代码：2.利用mask做GrabCut分割

///利用mask做GrabCut分割
//未做，待补

运行结果：待补

代码：3.漫水填充算法

///漫水填充算法
#include "opencv2/opencv.hpp"
using namespace cv;
Mat srcImg = imread("bird.jpg", CV_LOAD_IMAGE_COLOR);
Point seedPoint;//种子点, 漫水填充算法起始点
Scalar newVal = Scalar(rand() % 255, rand() % 255, rand() % 255);//像素点被染颜色的值, 即在重绘区域像素的新值
Scalar loDiff = Scalar(10, 10, 10);//待理解
Scalar upDiff = Scalar(10, 10, 10);//待理解
void onMouse(int event, int x, int y, int flags, void* param)
{
	switch (event)
	{
	case CV_EVENT_LBUTTONDOWN://鼠标左键按下
		seedPoint = Point(x, y);//此时位置即为种子点
		floodFill(srcImg, seedPoint, newVal, 0, loDiff, upDiff);// 漫水填充算法
		break;
	}
	imshow("floodFill", srcImg);//显示floodFill后的srcImg
}
int main()
{
	namedWindow("floodFill", CV_WINDOW_AUTOSIZE);//新建窗口
	setMouseCallback("floodFill", onMouse, 0);//设置鼠标回调函数，依附于"floodFill"窗口
	imshow("floodFill", srcImg);//使得一开始就显示"floodFill"窗口
	waitKey(0);
	return 0;
}

运行结果：