opencv学习-霍夫变换-直线检测

Posted 2021-08-02 殇堼

HoughLines

标准的霍夫变换 cv::HoughLines从平面坐标转换到霍夫空间，最终输出是

HoughLinesP

表示极坐标空间。
霍夫变换直线概率 cv::HoughLinesP最终输出是直线的两个点
cv::HoughLinesP(src，lines，rho，theta，minLineLength，maxLineGap）
InputArray src, // 输入图像必须8-bit图像（单通道），可以是灰度图像，或是二值化图像
OutputArray lines, // 输出的极坐标来表示直线
double rho, // 生成极坐标时候的像素扫描步长，为1。
double theta, //生成极坐标时候的角度步长，一般取值CV_PI/180
int threshold, // 阈值，只有获得足够交点的极坐标点才被看成是直线
double minLineLength=0;// 最小直线长度,小于该值得直线都会被去掉
double maxLineGap=0;// 最大间隔
)

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

using namespace std;
using namespace cv;

int main()
{
	Mat src, src_gray, bin,dst;
	src = imread("D:/images/lines.png");
	if (src.empty())
	{
		cout << "could not load image !";
		return -1;
	}
	imshow("输入图像", src);

	//canny处理之后图像已经就是灰度图像了
	Canny(src, src_gray, 0, 255);
	imshow("边缘检测", src_gray);

	//灰度图转回彩色图像
	cvtColor(src_gray,dst,COLOR_GRAY2BGR);
	
	//霍夫变换
	vector<Vec4f> plines;//霍夫变换得出的结果（直线）放入plines这个数组中。
	HoughLinesP(src_gray, plines, 1, CV_PI / 180, 150, 30, 10);//输入必须是单通道的图像，注意这里不能输入dst，而是src_gray
	cout << plines.size() << endl;//检测输出图像的位数是否为8位，单通道。
	Scalar color = Scalar(0, 0, 255);//颜色设置为红色。
	for (size_t i = 0; i < plines.size(); i++) {
		Vec4f hline = plines[i];
		//在dst彩色图像上绘制直线
		line(dst, Point(hline[0], hline[1]), Point(hline[2], hline[3]), color, 3, LINE_AA);
	}

	imshow("霍夫变换", dst);
	waitKey(0);
	return 0;
}

代码简化：

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

using namespace std;
using namespace cv;

int main()
{
	Mat src, src_gray, bin, dst;
	src = imread("D:/images/lines.png");
	if (src.empty())
	{
		cout << "could not load image !";
		return -1;
	}
	imshow("输入图像", src);

	//canny处理之后图像已经就是灰度图像了
	Canny(src, src_gray, 0, 255);
	imshow("边缘检测", src_gray);
	
	//霍夫变换
	vector<Vec4f> plines;//霍夫变换得出的结果（直线）放入plines这个数组中。
	HoughLinesP(src_gray, plines, 1, CV_PI / 180, 150, 30, 10);//输入必须是单通道的图像，注意这里不能输入dst，而是src_gray
	cout << plines.size() << endl;//检测输出图像的位数是否为8位，单通道。
	Scalar color = Scalar(0, 0, 255);//颜色设置为红色。
	for (size_t i = 0; i < plines.size(); i++) {
		Vec4f hline = plines[i];
	//在原图上绘制直线
		line(src, Point(hline[0], hline[1]), Point(hline[2], hline[3]), color, 3, LINE_AA);
	}

	imshow("霍夫变换", src);
	waitKey(0);
	return 0;
}