python - OpenCV houghLinesP 参数

Question

我很难在 Python 中使用 HoughLinesP 和 OpenCV 在此图像中找到棋盘上的线条。

为了理解HoughLinesP的参数，我想出了以下代码:

import numpy as np
import cv2
from matplotlib import pyplot as plt
from matplotlib import image as image

I = image.imread('chess.jpg') 
G = cv2.cvtColor(I, cv2.COLOR_BGR2GRAY)

# Canny Edge Detection:
Threshold1 = 150;
Threshold2 = 350;
FilterSize = 5
E = cv2.Canny(G, Threshold1, Threshold2, FilterSize)

Rres = 1
Thetares = 1*np.pi/180
Threshold = 1
minLineLength = 1
maxLineGap = 100
lines = cv2.HoughLinesP(E,Rres,Thetares,Threshold,minLineLength,maxLineGap)
N = lines.shape[0]
for i in range(N):
    x1 = lines[i][0][0]
    y1 = lines[i][0][1]    
    x2 = lines[i][0][2]
    y2 = lines[i][0][3]    
    cv2.line(I,(x1,y1),(x2,y2),(255,0,0),2)

plt.figure(),plt.imshow(I),plt.title('Hough Lines'),plt.axis('off')
plt.show()

我遇到的问题是这只会占用一行。如果我将 maxLineGap 减少到 1，它会增加数千。

我理解为什么会这样，但是我如何选择一组合适的参数来合并所有这些共线线？我错过了什么吗？

我想保持代码简单，因为我将它用作此函数的示例。

提前感谢您的帮助!

更新:这与 HoughLines 完美搭配。

而且似乎没有边缘检测问题，因为 Canny 工作得很好。

但是，我仍然需要让 HoughLinesP 工作。有什么想法吗？？

此处的图片:Results

Answer 1

好的，我终于找到了问题所在，并认为我会与其他为此疯狂的人分享解决方案。问题是在 HoughLinesP 函数中，有一个额外的参数“lines”，这是多余的，因为函数的输出是相同的:

cv2.HoughLinesP(image, rho, theta, threshold[, lines[, minLineLength[, maxLineGap]]])

这会导致参数出现问题，因为它们以错误的顺序读取。为了避免与参数的顺序混淆，最简单的解决方案是在函数中指定它们，如下所示:

lines = cv2.HoughLinesP(E,rho = 1,theta = 1*np.pi/180,threshold = 100,minLineLength = 100,maxLineGap = 50)

这完全解决了我的问题，希望对其他人有所帮助。