参照别人的霍夫变换直线检测,后面自己写实现代码

 

#include 
#include 
#include 

using namespace cv;
using namespace std;

int main()
{
	Mat srcImage = imread("city.jpg");
	imshow("Src Pic", srcImage);

	Mat midImage, dstImage;
	//边缘检测
	Canny(srcImage, midImage, 50, 250, 3);
	//灰度化
	cvtColor(midImage, dstImage, CV_GRAY2BGR);
	// 定义矢量结构存放检测出来的直线
	vector lines;
	//通过这个函数,我们就可以得到检测出来的直线集合了
	HoughLines(midImage, lines, 1, CV_PI / 180, 200, 0, 0);
	//这里注意第五个参数,表示阈值,阈值越大,表明检测的越精准,速度越快,得到的直线越少(得到的直线都是很有把握的直线)
	//这里得到的lines是包含rho和theta的,而不包括直线上的点,所以下面需要根据得到的rho和theta来建立一条直线

	//依次画出每条线段
	for (size_t i = 0; i < lines.size(); i++)
	{
		float rho = lines[i][0]; //就是圆的半径r
		float theta = lines[i][1]; //就是直线的角度
		Point pt1, pt2;
		double a = cos(theta), b = sin(theta);
		double x0 = a * rho, y0 = b * rho;
		pt1.x = cvRound(x0 + 1000 * (-b));
		pt1.y = cvRound(y0 + 1000 * (a));
		pt2.x = cvRound(x0 - 1000 * (-b));
		pt2.y = cvRound(y0 - 1000 * (a));

		line(dstImage, pt1, pt2, Scalar(55, 100, 195), 1, LINE_AA); //Scalar函数用于调节线段颜色,就是你想检测到的线段显示的是什么颜色

		imshow("边缘检测后的图", midImage);
		imshow("最终效果图", dstImage);
	}
	waitKey();
	return 0;
}

 

你可能感兴趣的:(参照别人的霍夫变换直线检测,后面自己写实现代码)