python - ハフ変換を使用して海洋画像の水平線を検出する際に、角度または線の許容誤差を設定する方法

Question

そのため、約 3000 枚の画像のデータセットがあり、そのうちの約 7% は処理が不十分です。私は OpenCV と Python を使用しています。私のアルゴリズムは、モルフォロジー侵食、キャニー エッジ検出、およびハフ ライン変換による前処理に基づいています。一部の画像では、水平線が検出されません。これは、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)