我很难在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
答案 0 :(得分:35)
好的,我终于找到了问题,并认为我会为其他任何人分享解决方案。问题是在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)
这完全解决了我的问题,我希望它能帮助别人。
答案 1 :(得分:2)
cv2.HoughLinesP(image,rho,theta,threshold,np.array([]),minLineLength = xx,maxLineGap = xx)
这也将起作用。
答案 2 :(得分:1)
import cv2
import numpy as np
img = cv2.imread('sudoku.png', cv2.IMREAD_COLOR)
# Convert the image to gray-scale
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# Find the edges in the image using canny detector
edges = cv2.Canny(gray, 50, 200)
# Detect points that form a line
lines = cv2.HoughLinesP(edges, 1, np.pi/180, 100, minLineLength=10, maxLineGap=250)
# Draw lines on the image
for line in lines:
x1, y1, x2, y2 = line[0]
cv2.line(img, (x1, y1), (x2, y2), (255, 0, 0), 3)
# Show result
img = cv2.resize(img, dsize=(600, 600))
cv2.imshow("Result Image", img)
if cv2.waitKey(0) & 0xff == 27:
cv2.destroyAllWindows()
答案 3 :(得分:0)
这不是HoughLinesP
问题,使用该方法只会获取图片中检测到的所有行并返回给您。
为了能够获得所需的线条,您需要在使用该方法之前平滑图像。但是,如果你太过平滑,那么HoughLinesP就无法检测到任何边缘。
您可以了解有关OpenCV here的平滑效果的更多信息。