OpenCV之霍夫变换

霍夫变换

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

cv.HoughLines(img, rho, theta, threshold)
参数：

img: 检测的图像，要求是二值化的图像，所以在调用霍夫变换之前首先要进行二值化，或者进行Canny边缘检测

rho、theta: \rhoρ 和\thetaθ的精确度

threshold: 阈值，只有累加器中的值高于该阈值时才被认为是直线。

import numpy as np
import random
import cv2 as cv
import matplotlib.pyplot as plt
# 1.加载图片，转为二值图
img = cv.imread('./hfxjc001.jpg')

gray = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
edges = cv.Canny(gray, 50, 150)

# 2.霍夫直线变换
lines = cv.HoughLines(edges, 0.8, np.pi / 180, 150)
# 3.将检测的线绘制在图像上（注意是极坐标噢）
for line in lines:
    rho, theta = line[0]
    a = np.cos(theta)
    b = np.sin(theta)
    x0 = a * rho
    y0 = b * rho
    x1 = int(x0 + 1000 * (-b))
    y1 = int(y0 + 1000 * (a))
    x2 = int(x0 - 1000 * (-b))
    y2 = int(y0 - 1000 * (a))
    cv.line(img, (x1, y1), (x2, y2), (0, 255, 0))
# # 4. 图像显示
# plt.figure(figsize=(10,8),dpi=100)
# plt.imshow(img[:,:,::-1]),plt.title('霍夫变换线检测')
# plt.xticks([]), plt.yticks([])
# plt.show()


# 4 图像显示
fig,axes=plt.subplots(nrows=2,ncols=2,figsize=(10,8))
axes[0,0].imshow(edges)
axes[0,0].set_title("二值图图")
axes[0,1].imshow(img[:,:,::-1]),plt.title('霍夫变换线检测')


plt.show()

x1 = int(x0 + 1000 * (-b))
y1 = int(y0 + 1000 * (a))

霍夫圆检测
API：

circles = cv.HoughCircles(image, method, dp, minDist, param1=100, param2=100, minRadius=0,maxRadius=0 )
参数：

image：输入图像，应输入灰度图像

method：使用霍夫变换圆检测的算法，它的参数是CV_HOUGH_GRADIENT

dp：霍夫空间的分辨率，dp=1时表示霍夫空间与输入图像空间的大小一致，dp=2时霍夫空间是输入图像空间的一半，以此类推

minDist为圆心之间的最小距离，如果检测到的两个圆心之间距离小于该值，则认为它们是同一个圆心

param1：边缘检测时使用Canny算子的高阈值，低阈值是高阈值的一半。

param2：检测圆心和确定半径时所共有的阈值

minRadius和maxRadius为所检测到的圆半径的最小值和最大值

返回：

circles：输出圆向量，包括三个浮点型的元素——圆心横坐标，圆心纵坐标和圆半径

import cv2 as cv
import numpy as np
import matplotlib.pyplot as plt
# 1 读取图像，并转换为灰度图
planets = cv.imread("./1124004.jpg")
gay_img = cv.cvtColor(planets, cv.COLOR_BGRA2GRAY)
# 2 进行中值模糊，去噪点
img = cv.medianBlur(gay_img, 7)  
# 3 霍夫圆检测
circles = cv.HoughCircles(img, cv.HOUGH_GRADIENT, 1, 200, param1=100, param2=30, minRadius=0, maxRadius=100)
# 4 将检测结果绘制在图像上
for i in circles[0, :]:  # 遍历矩阵每一行的数据
    # 绘制圆形
    cv.circle(planets, (i[0], i[1]), i[2], (0, 255, 0), 2)
    # 绘制圆心
    cv.circle(planets, (i[0], i[1]), 2, (0, 0, 255), 3)


# 4 图像显示
fig,axes=plt.subplots(nrows=2,ncols=2,figsize=(10,8))
axes[0,0].imshow(img)
axes[0,0].set_title("二值图图")
axes[0,1].imshow(planets[:,:,::-1]),plt.title('霍夫变换圆检测')


plt.show()