快捷導(dǎo)航

C++ Opencv自寫函數(shù)實現(xiàn)膨脹腐蝕處理技巧

更新時間：2021年10月13日 10:40:17 作者：秋沐霖

這篇文章主要介紹了C++ Opencv 自寫函數(shù)實現(xiàn)膨脹腐蝕處理,本文通過示例代碼給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下

一、膨脹腐蝕學(xué)習(xí)筆記

二、代碼及結(jié)果分享

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

using namespace std;
using namespace cv;

//定義腐蝕函數(shù)
void myErode(Mat Src, Mat Tem, Mat Dst)
{
	int m = (Tem.rows - 1) / 2;
	int n = (Tem.cols - 1) / 2;
	for (int i = m; i < Src.rows - m; i++)//i、j的范圍保證結(jié)構(gòu)元始終在擴展后的圖像內(nèi)部
	{
		for (int j = n; j < Src.cols - n; j++)
		{
			Mat SrcROI = Src(Rect(j - m, i - n, Tem.cols, Tem.rows));
			double sum = 0;
			sum = SrcROI.dot(Tem);//矩陣對應(yīng)位置相乘后求和
			if (sum == 9)//結(jié)構(gòu)元的9個元素均為1，和為9才能保證結(jié)構(gòu)元完全包含于相應(yīng)集合
				Dst.at<uchar>(i, j) = 255;
			else
				Dst.at<uchar>(i, j) = 0;
		}
	}
}

//定義膨脹函數(shù)
void myDilate(Mat Src, Mat Tem, Mat Dst)
{
	int m = (Tem.rows - 1) / 2;
	int n = (Tem.cols - 1) / 2;
	for (int i = m; i < Src.rows - m; i++)//i、j的范圍保證結(jié)構(gòu)元始終在擴展后的圖像內(nèi)部
	{
		for (int j = n; j < Src.cols - n; j++)
		{
			Mat SrcROI = Src(Rect(j - m, i - n, Tem.cols, Tem.rows));
			double sum = 0;
			sum = SrcROI.dot(Tem);//矩陣對應(yīng)位置相乘后求和
			if (sum != 0)//結(jié)構(gòu)元的9個元素均為1，只要和不為0，就能說明結(jié)構(gòu)元與相應(yīng)集合有交集
				Dst.at<uchar>(i, j) = 255;
			else
				Dst.at<uchar>(i, j) = 0;
		}
	}
}

int main()
{
	Mat mImage = imread("dada.jpg", 0);
	if (mImage.data == 0)
	{
		cerr << "Image reading error !" << endl;
		system("pause");
		return -1;
	}
	namedWindow("The original image", WINDOW_NORMAL);
	imshow("The original image", mImage);

	//設(shè)置閾值將圖像二值化
	const int binThreshold = 80;
	for (int i = 0; i < mImage.rows; i++)
	{
		uchar* pImage = mImage.ptr<uchar>(i);
		for (int j = 0; j < mImage.cols; j++)
		{
			if (pImage[j] > binThreshold)//事實上應(yīng)將灰度值大于閾值的賦值為255，但為了方便后續(xù)膨脹腐蝕的計算，在這里將其賦值為1
				pImage[j]= 1;
			else
				pImage[j] = 0;
		}
	}

	//定義并初始化結(jié)構(gòu)元
	Mat mTemplate(3, 3, CV_8UC1, Scalar(1));
	if (mTemplate.rows * mTemplate.cols % 2 == 0)
	{
		cerr << "The size doesn't meet the requirement !" << endl;
		system("pause");
		return -1;
	}
	
	//擴展圖像邊界
	copyMakeBorder(mImage, mImage, mTemplate.rows, mTemplate.rows, mTemplate.cols, mTemplate.cols, BORDER_CONSTANT, Scalar(0));

	//進行圖像腐蝕
	Mat mEResult = mImage.clone();
	myErode(mImage, mTemplate, mEResult);

	//進行圖像膨脹
	Mat mDResult = mImage.clone();
	myDilate(mImage, mTemplate, mDResult);

	//進行顯示
	namedWindow("The eroding image", WINDOW_NORMAL);
	imshow("The eroding image", mEResult);
	namedWindow("The dilating image", WINDOW_NORMAL);
	imshow("The dilating image", mDResult);
	waitKey();
	destroyAllWindows();

	return 0;
}

膨脹(dilate)和腐蝕(erode)均是對白色區(qū)域而言。由結(jié)果可明顯看出，膨脹后的白色面積要比腐蝕后的大。由圖像左下角的水印變化也可直觀看出兩種操作對圖像的不同影響。

三、注意事項

在本次編程實現(xiàn)過程中，為了確定結(jié)構(gòu)元是否包含于集合（或與集合是否有交集），需要讓結(jié)構(gòu)元中各元素與圖像中對應(yīng)位置像素值相乘后求和。Mat類型中有幾種不同類型的乘法，在這里加以總結(jié)。

Mat A, B ;

3.1A.dot(B)

A、B對應(yīng)位置元素相乘，之后將所有乘積相加求和，返回值是double型數(shù)字。要求A、B size必須相同。

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

using namespace std;
using namespace cv;

int main()
{
	uchar A[3][3] = { {1,2,3},{4,5,6},{7,8,9} };
	uchar B[3][3] = { { 1,1,1 },{ 1,1,1 },{ 1,1,1 } };
	Mat Src(3, 3, CV_8UC1, A);
	Mat Dst(3, 3, CV_8UC1, B);
	double Result = Src.dot(Dst);
	cout << "Src：" << Src << endl;
	cout << "Dst：" << Dst << endl;
	cout << "Result：" << Result << endl;
	system("pause");
	return 0;
}

3.2A.mul(B)

A、B對應(yīng)位置元素相乘，返回值是和A、B等大小，同類型的Mat型矩陣。要求A、B size必須相同。若計算結(jié)果溢出，則溢出值自動變?yōu)楫?dāng)前數(shù)據(jù)類型下允許的的最大值。

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

using namespace std;
using namespace cv;

int main()
{
	uchar A[3][3] = { {1,2,3},{4,5,6},{7,8,9} };
	uchar B[3][3] = { { 1,1,1 },{ 1,1,1 },{ 1,1,1 } };
	Mat Src(3, 3, CV_8UC1, A);
	Mat Dst(3, 3, CV_8UC1, B);
	Mat Result = Src.mul(Dst);
	cout << "Src：" << Src << endl;
	cout << "Dst：" << Dst << endl;
	cout << "Result：" << Result << endl;
	system("pause");
	return 0;
}