opencv python 直线检测_自学python-opencv(7)直线检测

霍夫变换(Hough Transform)是图像处理中的一种特征提取技术，它通过一种投票算法检测具有特定形状的物体。该过程在一个参数空间中通过计算累计结果的局部最大值得到一个符合该特定形状的集合作为霍夫变换结果。霍夫变换于1962年由Paul Hough 首次提出，后于1972年由Richard Duda和Peter Hart推广使用，经典霍夫变换用来检测图像中的直线，后来霍夫变换扩展到任意形状物体的识别，多为圆和椭圆。

原理介绍：

1、对于直角坐标系中的任意一点A(x0,y0)，经过点A的直线满足Y0=k*X0+b.(k是斜率，b是截距)

2、那么在X-Y平面过点A(x0,y0)的直线簇可以用Y0=k*X0+b表示，但对于垂直于X轴的直线斜率是无穷大的则无法表示。因此将直角坐标系转换到极坐标系就能解决该特殊情况。

3、在极坐标系中表示直线的方程为ρ=xCosθ+ySinθ(ρ为原点到直线的距离),如图所示：

如上图，假定在一个8*8的平面像素中有一条直线，并且从左上角(1,8)像素点开始分别计算θ为0°、45°、90°、135°、180°时的ρ，图中可以看出ρ分别为1、(9√2)/2、8、(7√2)/2、-1，并给这5个值分别记一票，同理计算像素点(3,6)点θ为0°、45°、90°、135°、180°时的ρ，再给计算出来的5个ρ值分别记一票，此时就会发现ρ = (9√2)/2的这个值已经记了两票了，以此类推，遍历完整个8*8的像素空间的时候ρ = (9√2)/2就记了5票， 别的ρ值的票数均小于5票，所以得到该直线在这个8*8的像素坐标中的极坐标方程为 (9√2)/2=x*Cos45°+y*Sin45°，到此该直线方程就求出来了。(PS：但实际中θ的取值不会跨度这么大，一般是PI/180)。

----------------------------------------------------------------------------------------------------------------------------------------------------------------

import cv2

import numpy

def line_detection(image):

gray=cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)

#apertureSize参数默认其实就是3

edges=cv2.Canny(gray,50,250,apertureSize=3)

'''

第一个参数，InputArray类型的image，输入图像，即源图像，需为8位的单通道二进制图像，可以将任意的源图载入进来后由函数修改成此格式后，再填在这里。

第二个参数，InputArray类型的lines，经过调用HoughLines函数后储存了霍夫线变换检测到线条的输出矢量。每一条线由具有两个元素的矢量表示，其中，是离坐标原点((0,0)(也就是图像的左上角)的距离。 是弧度线条旋转角度(0~垂直线，π/2~水平线)。

第三个参数，double类型的rho，以像素为单位的距离精度。另一种形容方式是直线搜索时的进步尺寸的单位半径。PS:Latex中/rho就表示 。

第四个参数，double类型的theta，以弧度为单位的角度精度。另一种形容方式是直线搜索时的进步尺寸的单位角度。

第五个参数，int类型的threshold，累加平面的阈值参数，即识别某部分为图中的一条直线时它在累加平面中必须达到的值。大于阈值threshold的线段才可以被检测通过并返回到结果中。

第六个参数，double类型的srn，有默认值0。对于多尺度的霍夫变换，这是第三个参数进步尺寸rho的除数距离。粗略的累加器进步尺寸直接是第三个参数rho，而精确的累加器进步尺寸为rho/srn。

第七个参数，double类型的stn，有默认值0，对于多尺度霍夫变换，srn表示第四个参数进步尺寸的单位角度theta的除数距离。且如果srn和stn同时为0，就表示使用经典的霍夫变换。否则，这两个参数应该都为正数。'''

lines=cv2.HoughLines(edges,1,numpy.pi/180,160)

#遍历lines数组

for line in lines:

print(type(lines))

rho,theta=line[0]

a=numpy.cos(theta)

b=numpy.sin(theta)

x0=a*rho

y0=b*rho

#这里的x1,y1,x2,y2是想通过角度theta画直线,1000这个值表示(x1,y1)(x2,y2)两点之间的距离为1000*2,也就是只要检测到图像中有直线,1000这个值是可以调整的.

x1=int(x0+1000*(-b))

y1=int(y0+1000*(a))

x2=int(x0-1000*(-b))

y2=int(y0-1000*(a))

cv2.line(image,(x1,y1),(x2,y2),(0,0,255),2)

cv2.imshow("line_detection",image)

def line_detect_possible(image):

gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# apertureSize参数默认其实就是3

edges = cv2.Canny(gray, 50, 250)

'''

第一个参数，InputArray类型的image，输入图像，即源图像，需为8位的单通道二进制图像，可以将任意的源图载入进来后由函数修改成此格式后，再填在这里。

第二个参数，InputArray类型的lines，经过调用HoughLinesP函数后后存储了检测到的线条的输出矢量，每一条线由具有四个元素的矢量(x_1,y_1, x_2, y_2) 表示，其中，(x_1, y_1)和(x_2, y_2) 是是每个检测到的线段的结束点。

第三个参数，double类型的rho， 以像素为单位的距离精度。 另一种形容方式是直线搜索时的进步尺寸的单位半径。

第四个参数，double类型的theta，以弧度为单位的角度精度。另一种形容方式是直线搜索时的进步尺寸的单位角度。

第五个参数，int类型的threshold，累加平面的阈值参数，即识别某部分为图中的一条直线时它在累加平面中必须达到的值。 大于阈值 threshold 的线段才可以被检测通过并返回到结果中。

第六个参数，double类型的minLineLength，有默认值0，表示最低线段的长度，比这个设定参数短的线段就不能被显现出来。

第七个参数，double类型的maxLineGap，有默认值0，允许将同一行点与点之间连接起来的最大的距离。

'''

lines = cv2.HoughLinesP(edges, 1, numpy.pi / 180, 100,minLineLength=20,maxLineGap=80)

for line in lines:

print(type(lines))

x1,y1,x2,y2=line[0]

cv2.line(image, (x1, y1), (x2, y2), (0, 0, 255), 2)

cv2.imshow("line_detect_possible", image)

print("-----------hi python-------------")

img=cv2.imread("./images/tianzige.jpg")

cv2.namedWindow("imput images",cv2.WINDOW_AUTOSIZE)

cv2.imshow("imput images",img)

line_detect_possible(img)

cv2.waitKey()

cv2.destroyWindow('all')

-------------------------------------------------------------------------------------------------------

---------------------------------------------------------------------------------------------------------------------------------------------------------

参考博客：https://blog.csdn.net/ycj9090900/article/details/52944708?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-3.add_param_isCf&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-3.add_param_isCf

官方参考：https://docs.opencv.org/4.1.0/d6/d10/tutorial_py_houghlines.html