OpenCV 中如何进行霍夫变换？

推荐答案

在OpenCV中，霍夫变换（Hough Transform）主要用于检测图像中的直线或圆等几何形状。以下是使用OpenCV进行霍夫变换的步骤：

1. 霍夫直线变换

// www.javascriptcn.com code example
import cv2
import numpy as np

# 读取图像并转换为灰度图
image = cv2.imread('image.jpg')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# 使用Canny边缘检测
edges = cv2.Canny(gray, 50, 150, apertureSize=3)

# 进行霍夫直线变换
lines = cv2.HoughLines(edges, 1, np.pi/180, 200)

# 绘制检测到的直线
if lines is not None:
    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))
        cv2.line(image, (x1, y1), (x2, y2), (0, 0, 255), 2)

# 显示结果
cv2.imshow('Hough Lines', image)
cv2.waitKey(0)
cv2.destroyAllWindows()

2. 霍夫圆变换

// www.javascriptcn.com code example
import cv2
import numpy as np

# 读取图像并转换为灰度图
image = cv2.imread('image.jpg', cv2.IMREAD_COLOR)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# 使用高斯模糊减少噪声
gray = cv2.medianBlur(gray, 5)

# 进行霍夫圆变换
circles = cv2.HoughCircles(gray, cv2.HOUGH_GRADIENT, 1, 20,
                           param1=50, param2=30, minRadius=0, maxRadius=0)

# 绘制检测到的圆
if circles is not None:
    circles = np.uint16(np.around(circles))
    for i in circles[0, :]:
        # 绘制外圆
        cv2.circle(image, (i[0], i[1]), i[2], (0, 255, 0), 2)
        # 绘制圆心
        cv2.circle(image, (i[0], i[1]), 2, (0, 0, 255), 3)

# 显示结果
cv2.imshow('Hough Circles', image)
cv2.waitKey(0)
cv2.destroyAllWindows()

本题详细解读

霍夫直线变换

霍夫直线变换是一种用于检测图像中直线的技术。其基本原理是将图像空间中的直线转换到参数空间（极坐标空间），通过累加器数组来检测直线。

• cv2.HoughLines()：该函数用于检测图像中的直线。参数包括：

  • edges：输入的二值图像，通常是Canny边缘检测的结果。
  • rho：距离分辨率，单位为像素。
  • theta：角度分辨率，单位为弧度。
  • threshold：累加器阈值，只有大于该值的直线才会被检测到。

• 绘制直线：通过极坐标参数（rho, theta）计算出直线的两个端点，并在图像上绘制出来。

霍夫圆变换

霍夫圆变换是一种用于检测图像中圆的技术。其基本原理是通过累加器数组来检测圆的中心点和半径。

• cv2.HoughCircles()：该函数用于检测图像中的圆。参数包括：

  • gray：输入的灰度图像。
  • method：检测方法，通常使用cv2.HOUGH_GRADIENT。
  • dp：累加器分辨率与图像分辨率的比值。
  • minDist：检测到的圆之间的最小距离。
  • param1：Canny边缘检测的高阈值。
  • param2：累加器阈值，只有大于该值的圆才会被检测到。
  • minRadius 和 maxRadius：检测到的圆的最小和最大半径。

• 绘制圆：通过检测到的圆的中心点和半径，在图像上绘制出圆和圆心。