霍夫直线检测原理

在CV中,经常需要检测直线,用到霍夫变换算法。(感觉自己的语言表达能力有所欠缺,所以写下来)

在图像空间中,可以用y=mx+b表示一条直线。其中,x、y为变量,m、b为参数

对其进行变换,b=-mx+y。这样,就完成了图像空间到参数空间的转换

从图像空间到参数空间,一条直线变成了一个点(m,b)

对于图像空间中的一个点,图像空间中的每条经过该点的直线都对应着参数空间中的一个点(m,b),这些点在参数空间中构成一条直线

 * 当图像中有两个点时,他们在参数空间中对应着两条必相交的直线,直线的交点(m,b)对应着图像空间中过着两点的直线

* 当图像空间中有多个点时,参数空间中相应的会有多条直线;当图像空间中的点在一条直线上时,参数空间中的多条直线会相交于同一点(m,b),即为图像空间中直线的斜率和截距。霍夫直线检测算法就是基于该原理。

*  因此,霍夫直线检测算法总结如下:

1)参数空间量化

2)创建参数空间累加器A(m,b),并初始化为0

3)对于图像空间直线上的每个点:

         对应着参数空间中的一个直线,若参数空间的点(m,b)在该直线上,那么A(m,b)加1

4)找到A(m,b)的最大值,即为直线的斜率和截距

for point in image_lines:    #图像空间中的point对应参数空间的parameters_line(b=-xm+y)
    for m,b in A:
        if m,b in parameters_line:
            A[m,b]+=1

res=max(A(m,b))

* 然而,参数空间的范围太大了,因此使用极坐标:

  

  此时,图像空间和参数空间的对应关系如下:

  

  从图中可以看出,霍夫直线检测即为在参数空间中对rho和theta投票的过程,得票最高者为最终的直线参数。

import cv2 as cv import numpy as np #直线检测 #使用霍夫直线变换做直线检测,前提条件:边缘检测已经完成 #标准霍夫线变换 def line_detection(image): gray = cv.cvtColor(image, cv.COLOR_RGB2GRAY) edges = cv.Canny(gray, 50, 150) #apertureSize参数默认其实就是3 cv.imshow("edges", edges) #cv.HoughLines参数设置:参数1,灰度图像;参数二,以像素为单位的距离精度(一般都是1,进度高,但是速度会慢一点) #参数三,以弧度为单位的角度精度(一般是1rad);参数四,阈值,大于阈值threshold的线段才可以被检测通过并返回到结果中 #该函数返回值为rho与theta lines = cv.HoughLines(edges, 1, np.pi/180, 200) for line in lines: rho, theta = line[0] #line[0]存储的是点到直线的极径和极角,其中极角是弧度表示的。 a = np.cos(theta) #theta是弧度 b = np.sin(theta) x0 = a * rho #代表x = r * cos(theta) y0 = b * rho #代表y = r * sin(theta) x1 = int(x0 + 1000 * (-b)) #计算直线起点横坐标 y1 = int(y0 + 1000 * a) #计算起始起点纵坐标 x2 = int(x0 - 1000 * (-b)) #计算直线终点横坐标 y2 = int(y0 - 1000 * a) #计算直线终点纵坐标 注:这里的数值1000给出了画出的线段长度范围大小,数值越小,画出的线段越短,数值越大,画出的线段越长 cv.line(image, (x1, y1), (x2, y2), (0, 0, 255), 2) #点的坐标必须是元组,不能是列表。 cv.imshow("image-lines", image) #统计概率霍夫线变换 def line_detect_possible_demo(image): gray = cv.cvtColor(image, cv.COLOR_RGB2GRAY) edges = cv.Canny(gray, 50, 150, apertureSize=3) # apertureSize参数默认其实就是3 lines = cv.HoughLinesP(edges, 1, np.pi / 180, 60, minLineLength=60, maxLineGap=5) for line in lines: x1, y1, x2, y2 = line[0] cv.line(image, (x1, y1), (x2, y2), (0, 0, 255), 2) cv.imshow("line_detect_possible_demo",image) src = cv.imread("E:/opencv/picture/track.jpg") print(src.shape) cv.namedWindow('input_image', cv.WINDOW_AUTOSIZE) cv.imshow('input_image', src) line_detection(src) src = cv.imread("E:/opencv/picture/track.jpg") #调用上一个函数后,会把传入的src数组改变,所以调用下一个函数时,要重新读取图片 line_detect_possible_demo(src) cv.waitKey(0) cv.destroyAllWindows() 霍夫检测直线原理: 关于hough变换,核心以及难点就是关于就是有原始空间到参数空间的变换上。以直线检测为例,假设有一条直线L,原点到该直线的垂直距离为p,垂线与x轴夹角为θθ,那么这条直线是唯一的,且直线的方程为 ρ=xcosθ+ysinθρ=xcosθ+ysinθ, 如下图所
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值