OpenCV利用霍夫变换进行直线检测

时间:2021-05-20

本文实例为大家分享了OpenCV利用霍夫变换进行直线检测的具体代码,供大家参考,具体内容如下

1.最简单的霍夫变换是在图像中识别直线。在平面直角坐标系(x-y)中,一条直线可以用下式表示:y=kx+b。

这表示参数平面(k-b)中的一条直线。因此,图像中的一个点对应参数平面中的一条直线,图像中的一条直线对应参数平面中的一个点。对图像上所有的点作霍夫变换,最终所要检测的直线对应的一定是参数平面中直线相交最多的那个点。这样就在图像中检测出了直线。在实际应用中,直线通常采用参数方程:p=x\cos\theta+y\sin\theta。

Opencv里有以下函数检测直线(最基本的霍夫变换):

void HoughLines(InputArray image, OutputArray lines, double rho, double theta, int threshold, double srn=0, double stn=0 )

具体用法看代码就知道了:(现在版本的OpenCV使用函数cvHoughLines2)

#include "opencv2/opencv.hpp"#define PI 3.1415926 int main(int argc, char *argv[]){ cv::Mat image = cv::imread ("road.jpg"); cv::Mat result; cv::cvtColor (image,result,CV_BGRA2GRAY); cv::Mat contours; cv::Canny (result,contours,125,350); //边缘检测 std::vector<cv::Vec2f> lines; cv::HoughLines (contours,lines,1,PI/180,80); std::vector<cv::Vec2f>::const_iterator iter = lines.begin (); std::cout<<lines.size ()<<std::endl; while(iter != lines.end()) { float rho = (*iter)[0]; float theta = (*iter)[1]; if(theta<PI/4.||theta>3.*PI/4) { //画交点在上下两边的直线 cv::Point pt1(rho/cos(theta),0); cv::Point pt2((rho-result.rows*sin(theta))/cos(theta),result.rows); cv::line(image,pt1,pt2,cv::Scalar(255),1); } else { //画交点在左右两边的直线 cv::Point pt1(0,rho/sin(theta)); cv::Point pt2(result.cols,(rho-result.cols*cos(theta)/sin(theta))); cv::line(image,pt1,pt2,cv::Scalar(255),1); } ++iter; } cv::namedWindow ("hough"); cv::imshow("hough",image); cv::waitKey (0);}

测试结果如下:

2.可以看出,上面的直线检测存在以下问题:

1)只能检测出线段所在的直线,而不知道具体线段位置,也不知道线段长度;

2)同一直线可能检测出多条直线;

3)偶然地也可能误判段直线。

针对这些问题,opencv有那么一个函数:(现在版本的OpenCV使用同一个函数cvHoughLines2)

void HoughLinesP(InputArray image, OutputArray lines, double rho, double theta, int threshold, dou-
ble minLineLength=0, double maxLineGap=0)

这个方法是通过概率霍夫变换实现的:

1)随机获取边缘图片上的前景点,映射到级坐标系画曲线;

2)当极坐标系里面有交点达到最小投票数,将该点对应x-y坐标系的直线L找出来;

3)搜索边缘图片上前景点,在直线L上的点(且点与点之间距离小于maxLineGap的)连成线段,然后这些点全部删除,并且记录该线段的参数,就是起始点和终止点。(当然这里线段长度要满足最小长度的,否则就不用记录了)

4)重复1),2),3)

其使用方法见代码:

#include "opencv2/opencv.hpp"#define PI 3.1415926 class LineFinder{private: std::vector<cv::Vec4i> lines; // 直线对应的点参数向量 double deltaRho; //步长 double deltaTheta; int minVote; // 判断是直线的最小投票数 double minLength; // 判断是直线的最小长度 double maxGap; // 同一条直线上点之间的距离容忍度public: LineFinder() : deltaRho(1), deltaTheta(PI/180), minVote(10), minLength(0.), maxGap(0.) {} //初始化 void setAccResolution(double dRho, double dTheta) // 设置步长 { deltaRho= dRho; deltaTheta= dTheta; } void setMinVote(int minv) // 设置最小投票数 { minVote= minv; } void setLineLengthAndGap(double length, double gap) // 设置最小线段长度和线段间距容忍度 { minLength= length; maxGap= gap; } std::vector<cv::Vec4i> findLines(cv::Mat& binary) //寻找线段 { lines.clear(); cv::HoughLinesP(binary,lines, deltaRho, deltaTheta, minVote,minLength, maxGap); return lines; } void drawDetectedLines(cv::Mat &image, cv::Scalar color=cv::Scalar(255,255,255)) // 画线段 { std::vector<cv::Vec4i>::const_iterator it2=lines.begin(); while (it2!=lines.end()) { cv::Point pt1((*it2)[0],(*it2)[1]); cv::Point pt2((*it2)[2],(*it2)[3]); cv::line( image, pt1, pt2, color); ++it2; } }}; int main(int argc, char *argv[]){ cv::Mat image = cv::imread ("road.jpg"); cv::Mat result; cv::cvtColor (image,result,CV_BGRA2GRAY); cv::Mat contours; cv::Canny (result,contours,125,350); //边缘检测 LineFinder finder; finder.setMinVote (80); finder.setLineLengthAndGap (100,20); finder.findLines (contours); finder.drawDetectedLines (image); std::vector<cv::Vec2f> lines; cv::HoughLines (contours,lines,1,PI/180,80); std::vector<cv::Vec2f>::const_iterator iter = lines.begin (); std::cout<<lines.size ()<<std::endl; while(iter != lines.end()) { float rho = (*iter)[0]; float theta = (*iter)[1]; if(theta<PI/4.||theta>3.*PI/4) { //画交点在上下两边的直线 cv::Point pt1(rho/cos(theta),0); cv::Point pt2((rho-result.rows*sin(theta))/cos(theta),result.rows); cv::line(image,pt1,pt2,cv::Scalar(255),1); } else { //画交点在左右两边的直线 cv::Point pt1(0,rho/sin(theta)); cv::Point pt2(result.cols,(rho-result.cols*cos(theta)/sin(theta))); cv::line(image,pt1,pt2,cv::Scalar(255),1); } ++iter; } cv::namedWindow ("hough"); cv::imshow("hough",image); cv::waitKey (0);}

测试结果如下:

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章