python - 如何在使用霍夫变换检测海洋图像上的水平线时设置一些角度或线容差

Question

所以我有一个包含大约 3000 张图像的数据集，其中大约 7% 的图像处理不当。我正在使用 OpenCV 和 Python，我的算法基于形态侵蚀、Canny 边缘检测和霍夫线变换的预处理。在某些图像上未检测到水平线，因为 Canny 返回一条有间隙的线或不是直线的水平线。有没有办法解决它？也许增加角度的公差或类似的东西？概率霍夫变换也无济于事。这是一个原始图像并成功处理了一个： 这是一个有问题的对：

import cv2
import numpy as np
img = cv2.imread(conf.data_folder + 'frame0134.jpg',0)
kernel = np.ones((9,9),np.uint8)
erosion = cv2.morphologyEx(img, cv2.MORPH_CLOSE, kernel)
edges = cv2.Canny(erosion,50,150)
lines = cv2.HoughLines(edges,1,np.pi/180,80)
for ro, theta in lines[0]:
  if theta == 0.0:
    x1 = int(ro)
    x2 = int(ro)
    y1 = 0
    y2 = img.shape[0]
    cv2.line(img,(x1,y1),(x2,y2),(0,0,255),2)
    cropped = img[:,0:x1]
  else:
    a = -1/np.tan(theta)
    b = ro/np.sin(theta)
    y1 = 0 x1 = max(0, int( -b/a))
    y2 = img.shape[0]
    x2 = min(img.shape[1], int((y2-b)/a))
    cv2.line(img,(x1,y1),(x2,y2),(0,0,255),2)