OpenCV-Python：霍夫变换

霍夫变换常用来在图像中提取直线和圆等几何形状。如下图：

我们下面来看看如何使用霍夫变换来检测直线。一条直线可以用数学表达式 y = mx + 或者 ρ = xcosθ + y sinθ表示（极坐标）

简单说明一下：

ρ 是从原点到直线的垂直距离，θ是直线的垂线与横轴顺时针方向的夹角，如下图所示：

首先创建一个2D数组（累加器），初始化累加器，所有的值都为0。行表示 ρ，列表示 θ。这个数组的大小决定了最后结果的准确性。如果你希望角度精确到1°，你就需要180列。对于 ρ，最大值为图片对角线距离。

想象一下我们有一个大小为100x100的直线位于图像中央。取直线上的第一个点，我们知道此处的（x，y）值，把x和y带入公式：ρ = xcosθ + y sinθ，然后遍历 θ 的取值0， 1， 2， 3，...，180.分别求出对应的 ρ 值，这样我们就得到了一系列（ρ，θ）的数值对，如果这个数值对在累加器中也存在相应的位置，就在这个位置上加 1。由于同一条直线上的点必然会有同样的（ρ，θ）。所以现在累加器中的（50，90）=1。现在取直线上的第二个点。重复上边的过程。更新累加器中的值。现在累加器中（50,90）的值为 2。你每次做的就是更新累加器中的值。对直线上的每个点都执行上边的操作，每次操作完成之后，累加器中的值就加 1，但其他地方有时会加 1, 有时不会。按照这种方式下去，到最后累加器中（50,90）的值肯定是最大的。如果你搜索累加器中的最大值，并找到其位置（50,90），这就说明图像中有一条直线，这条直线到原点的距离为 50，它的垂线与横轴的夹角为 90 度。

OpenCV中首先计算(r,θ) 累加数，累加数超过一定值后就认为在同一直线上（有一个阈值）。

霍夫直线变换

import cv2

import numpy as np

# .加载图片，转为二值图

img = cv2.imread('shapes.jpg')

drawing = np.zeros(img.shape[:], dtype=np.uint8)

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

edges = cv2.Canny(gray, , )

# .霍夫直线变换

lines = cv2.HoughLines(edges, 0.8, np.pi / , )

函数中：

参数1：要检测的二值图（一般是阈值分割或边缘检测后的图）
参数2：距离 ρ 的精度，值越大，考虑越多的线
参数3：角度 θ 的精度，值越小，考虑越多的线
参数4：累加数阈值，值越小，考虑越多的线

# .将检测的线画出来（注意是极坐标噢）

for line in lines:

    rho, theta = line[]

    a = np.cos(theta)

    b = np.sin(theta)

    x0 = a * rho

    y0 = b * rho

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

    y1 = int(y0 +  * (a))

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

    y2 = int(y0 -  * (a))

    cv2.line(drawing, (x1, y1), (x2, y2), (, , ))

统计概率霍夫直线变换

前面的方法又称为标准霍夫变换，它会计算图像中的每一个点，计算量比较大，另外它得到的是整一条线，并不知道原图中直线的端点。所以提出了概率霍夫直线变换（Probabilistic Hough Transform），是一种改进的霍夫变换：

drawing = np.zeros(img.shape[:], dtype=np.uint8)

# 统计概率霍夫线变换

lines = cv2.HoughLinesP(edges, 0.8, np.pi / , , minLineLength=, maxLineGap=)


# 将检测的线画出来

for line in lines:

    x1, y1, x2, y2 = line[]

    cv2.line(drawing, (x1, y1), (x2, y2), (, , ), , lineType = cv2.LINE_AA)

cv2.imwrite('houghlines4.jpg', drawing)

参数：

minLineLength：最短长度阈值，比这个长度短的线会被排除
maxLineGap：同一直线两点之间的最大距离

cv2.LINE_AA：抗锯齿线型

霍夫圆变换

霍夫圆变换跟直线变换类似，只不过线是用（r，θ），圆是用(x_center,y_center,r)来表示，从二维变成了三维，数据量变大了很多；所以一般使用霍夫梯度法去减少计算量

drawing = np.zeros(img.shape[:], dtype=np.uint8)

# 霍夫圆变换

circles = cv2.HoughCircles(edges, cv2.HOUGH_GRADIENT, , , param2=)

circles = np.int0(np.around(circles))

# 将检测的圆画出来

for i in circles[, :]:

    cv2.circle(drawing, (i[], i[]), i[], (, , ), )  # 画出外圆

    cv2.circle(drawing, (i[], i[]), , (, , ), )  # 画出圆心

cv2.imwrite('HoughCircles.jpg', drawing)

参数2：变换方法，一般使用霍夫梯度法，详情：HoughModes
参数3 dp=1：表示霍夫梯度法中累加器图像的分辨率与原图一致
参数4：两个不同圆圆心的最短距离
参数5：param2跟霍夫直线变换中的累加数阈值一样