【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;
}

運行結果：