通过 Canny 和 Hough 进行地平线检测

Question

我正在尝试在 python 中检测水平线。为了实现我的目标，我首先使用了精明的边缘检测算法。 sea.jpg

import cv2
import numpy as np

gray = cv2.imread('images/sea.jpg')
edges = cv2.Canny(gray,50,150,apertureSize = 3)

边缘检测后，我对图像应用霍夫变换。

minLineLength=100
lines = cv2.HoughLinesP(image=edges,rho=10,theta=np.pi/180, 
threshold=1,lines=np.array([]), minLineLength=minLineLength,maxLineGap=80)

a,b,c = lines.shape
for i in range(a):
    cv2.line(gray, (lines[i][0][0], lines[i][0][1]), (lines[i][0][2], lines[i][0][3]), (0, 0, 255), 3, cv2.LINE_AA)
    cv2.imwrite('images/output.jpg',gray)

cv2.imshow('output',gray)
cv2.waitKey(0)
cv2.destroyAllWindows()

但最后，我output.jpg 喜欢这样。

我做错了什么？

Answer 1

您需要过滤掉 theta 值较低的行。我认为您目前没有过滤任何内容。尝试创建一个新的行列表，其中仅包含 y1 == y2 的行，即水平线。