C++ Opencv HoughLines()用霍夫变换在二元图像中寻线

一、霍夫变换简介

参考http://homepages.inf.ed.ac.uk/rbf/HIPR2/hough.htm

二、HoughLines()函数详解

该函数接受的输入矩阵只能是8位单通道的二值化灰度图。

三、代码及结果分享

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

using namespace std;
using namespace cv;

void myHough(Mat src, Mat dst)
{
	vector<Vec2f> lines;//用于储存参数空间的交点
	HoughLines(src, lines, 1, CV_PI / 180, 120, 0, 0);//针对不同像素的图片注意调整阈值
	const int alpha = 1000;//alpha取得充分大，保证画出贯穿整个图片的直线
	//lines中存储的是边缘直线在极坐标空间下的rho和theta值，在图像空间(直角坐标系下)只能体现出一个点
	//以该点为基准，利用theta与斜率之间的关系，找出该直线上的其他两个点(可能不在图像上)，之后以这两点画出直线
	for (size_t i = 0; i < lines.size(); i++)
	{
		float rho = lines[i][0], theta = lines[i][1];
		double cs = cos(theta), sn = sin(theta);
		double x = rho * cs, y = rho * sn;
		Point pt1(cvRound(x + alpha * (-sn)), cvRound(y + alpha * cs));
		Point pt2(cvRound(x - alpha * (-sn)), cvRound(y - alpha * cs));
		line(dst, pt1, pt2, Scalar(0, 0, 255), 1, LINE_AA);
	}
}

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

	Mat mMiddle ;
	cvtColor(mImage, mMiddle,  COLOR_BGR2GRAY);//Canny()只接受单通道8位图像，边缘检测前先将图像转换为灰度图
	Canny(mImage, mMiddle, 50, 150, 3);//边缘检测，检测结果作为霍夫变换的输入
	Mat mResult = mImage.clone();
	myHough(mMiddle, mResult);//将结果展示在原图像上
	namedWindow("The processed image", WINDOW_NORMAL);
	imshow("The processed image", mResult);
	waitKey();
	destroyAllWindows();
	return 0;
}