Hough 线变换未正确识别任何线

Question

我在霍夫线变换方面遇到了问题。我正在尝试确定厨房中的主要线路。我首先只使用了 Canny，但它接收到的噪音比我想要的要多，而且没有接收到墙壁和天花板的交汇处。但是，霍夫线变换仅识别一条它根本不应该识别的线。任何帮助将不胜感激。

我的输入：

kitchen_sample.jpg

我的输出：

kitchen_lines.jpg

这是我的代码：

import cv2
import numpy as np

image = cv2.imread('kitchen_sample.jpg')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
edges = cv2.Canny(gray, 50, 150, apertureSize=3)

lines = cv2.HoughLines(edges, 1, np.pi / 180, 200)
for rho, theta in lines[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.imwrite('kitchen_lines.jpg', image)

Answer 1

您可能正在查看旧的 opencv 教程页面，其中可能有错误（或者版本控制发生了变化，没有跟踪 opencv-python）。
Here's a new & correct one

您只需更换

for rho, theta in lines[0]:

和

for line in lines:
rho,theta = line[0]

但无论如何，您需要一些时间才能获得所需的输出。我建议您使用 HoughLinesP，这很容易为您提供您可能需要的东西

lines = cv2.HoughLinesP(edges,1,np.pi/180,100,minLineLength=100,maxLineGap=10)
for line in lines:
x1,y1,x2,y2 = line[0]
cv2.line(image,(x1,y1),(x2,y2),(0,255,0),2)

Hough 线变换未正确识别任何线

Hough Line transform is not correctly identifying any lines

python

opencv

hough-transform

canny-operator