【數字圖像處理】自實作Harris角點檢測

關于Harris角點檢測的原理網上已經很多了，在此不再贅述，本文主要是基于原理自己實作了一下Harris的算法，權當留個筆記，如果能對看到此文的朋友有所幫助，甚感欣慰。

廢話不再多說，直接上代碼。

自定義的Harris方法：

void Harris(Mat src, Mat& dst, int blockSize, int sobelKSize = 3)
{
	Mat Dx = Mat(src.size(), src.type());//建立矩陣用以儲存圖像在x方向上的一階差分
	Mat Dy = Mat(src.size(), src.type());//建立矩陣用以儲存圖像在y方向上的一階差分

	if(sobelKSize > 0)
	{
		Sobel(src, Dx, CV_32F, 1, 0, sobelKSize, 1, 0, 0);//計算圖像在x方向上的一階差分dx
		Sobel(src, Dy, CV_32F, 0, 1, sobelKSize, 1, 0, 0);//計算圖像在y方向上的一階差分dy
	}
	else
	{
		//當輸入參數sobelKSize <= 0時，采用預設3*3的sobel核心對圖像進行卷積
		//openCV源碼中使用的函數為	Scharr(src, dst, CV_32F, 1, 0); Scharr(src, dst, CV_32F, 0, 1);
		//至于為啥用scharr，官方解釋為，當sobel算子的核心大小為3*3時，其計算可能産生一定的誤差，是以openCV提供了一個更為準确的3*3核心的一階差分算子scharr
		Sobel(src, Dx, CV_32F, 1, 0);
		Sobel(src, Dy, CV_32F, 0, 1);
	}

	//建立3通道矩陣cov，用以儲存每個像素對應的dx*dx，dx*dy，dy*dy
	Mat cov = Mat(src.size(), CV_32FC3);

	for(int i = 0;i < src.rows;i++)
	{
		float* data = cov.ptr<float>(i);
		float* dxData = Dx.ptr<float>(i);
		float* dyData = Dy.ptr<float>(i);
		for(int j = 0;j < src.cols;j++)
		{
			float dx = dxData[j];
			float dy = dyData[j];
			data[3 * j] = dx * dx;
			data[3 * j + 1] = dx * dy;
			data[3 * j + 2] = dy * dy;
		}
	}
	//對計算得到的協方差矩陣元素進行高斯權重。openCV源碼中使用的函數為boxFilter
	if(blockSize > 0)
		GaussianBlur(cov, cov, Size(blockSize,blockSize), 2.0, 0);
	else
		GaussianBlur(cov, cov, Size(3,3), 2.0, 0);
// 	boxFilter(cov,+ cov, cov.depth(), Size(3, 3),
// 		Point(-1,-1), true, 0 );

	//通過公式計算R，R = det(M) - k * trace(M)^2，R值越大越有可能是角點
	//k取經驗值，一般取0.04-0.06
	for(int i = 0;i < dst.rows;i++)
	{
		float* data = dst.ptr<float>(i);
		float* covData = cov.ptr<float>(i);
		for(int j = 0;j < dst.cols;j++)
		{
			float a = covData[j * 3];
			float b = covData[j * 3 + 1];
			float c = covData[j * 3 + 2];

			data[j] = (float)(a * c - b * b - 0.05 * (a + c) * (a + c));
		}
	}
}
           

int _tmain(int argc, _TCHAR* argv[])
{
	Mat src = imread("E:\\學習之路\\數字圖像處理練習程式\\004.png", 0);
	Mat dst = Mat(src.size(), CV_32F);

	Harris(src, dst, 3, 3);

	normalize(dst, dst, 1.0, 0, NORM_MINMAX);

	imwrite("E:\\學習之路\\數字圖像處理練習程式\\xiaoguo.jpg", dst);

	imshow("111", dst);
	waitKey();

	return 0;
}
           

最終的角點檢測效果圖為（歸一化之後）:

原圖像為：