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

Question

我正在尝试检测python中的地平线。为了实现我的目标，我首先使用了精明的边缘检测算法。海.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()

但最后，我输出.jpg像这样。

我做错了什么？

Answer 1

您需要

过滤掉θ值较低的线。我认为您目前没有过滤任何东西。尝试创建一个新的线条列表，其中只有 y1 == y2 的行，换句话说，水平线。