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

Question

我正在尝试检测python中的水平线。为了实现我的目标，首先我使用了canny边缘检测算法。 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的行，换句话说是水平行。