Python OpenCV HoughLinesP不准确的直线检测

Question

考虑图像中的三个点A，B，C。

​

​

下面是它们在300x300大小图像中的坐标。

​

​

我试图使用下面的HoughLinesP代码检测并绘制一条通过这三个点连接起来的线。

import cv2
import numpy as np

img = cv2.imread('test.png')
img = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY) #Convert img to grayscale 
lines = cv2.HoughLinesP(img, rho=1, theta=np.pi/180, threshold=1, minLineLength=5, maxLineGap=10)
print(lines)

for line in lines:
    x1, y1, x2, y2 = line[0]
    cv2.line(img, (x1, y1), (x2, y2), 255, 1)
cv2.imshow("result", img)

但是它检测到一条只穿过B和C的线，为什么会这样呢？

Output:
[[[110 100 120 100]]]

​

​

Answer 1

• 增加累加器的分辨率，为rho和θ提供更小的值。
• 将累加器的阈值设置为最小2。

lines = cv2.HoughLinesP(img, rho=0.1, theta=np.pi/180 * 0.1, threshold=2, minLineLength=5, maxLineGap=10)

​

​

Answer 2

cv2.HoughLinesP()主要用于检测一条直线，而不是真正用于绘图。考虑到你的三点，你还可以尝试一些其他的选择。第一种方法是通过查找最左边和最右边的点来过滤点，然后用cv2.line()绘制直线。另一种方法是找到所有的点，然后使用cv2.fillPoly()。第三种方法是使用cv2.polylines()

​

​

import cv2
import numpy as np

image = cv2.imread('1.png')
gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)
coords = np.column_stack(np.where(gray > 0))
coords = np.rot90(coords, 2)
cv2.fillPoly(image, [coords], (36,255,12)) # or
# cv2.polylines(image, [coords], 1, (36,255,12))

cv2.imshow('image', image)
cv2.imwrite('image.png', image)
cv2.waitKey()