opencv ハフ変換 横線だけを検出

Question

opencvのハフ変換で横線だけを検出するにはどうすればいいですか？
このような画像に対して：

このような出力がほしいです：

Answer 1

OpenCV: Feature Detection

ハフ変換の返り値として、直線を表すパラメータ (rho, theta) のリストが得られるので、

lines = list(filter(lambda x: abs(x[1] - np.pi / 2) <= np.deg2rad(5), lines))

として、theta が 90° 付近のものだけ抽出すればよいと思います。
角度はラジアンなので注意してください。

import cv2
import numpy as np

img = cv2.imread("sample.jpeg")

gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

edges = cv2.Canny(gray, 150, 300, L2gradient=True)

# ハフ変換で直線検出する。
lines = cv2.HoughLines(edges, 1, np.pi / 180, 200)
lines = lines.squeeze(axis=1)

# 直線の角度がほぼ90°(-85° ~ 95°) のものだけ抽出する。
lines = list(filter(lambda x: abs(x[1] - np.pi / 2) <= np.deg2rad(5), lines))


# 直線を描画する。
def draw_line(img, theta, rho):
    h, w = img.shape[:2]
    if np.isclose(np.sin(theta), 0):
        x1, y1 = rho, 0
        x2, y2 = rho, h
    else:
        calc_y = lambda x: rho / np.sin(theta) - x * np.cos(theta) / np.sin(theta)
        x1, y1 = 0, int(calc_y(0))
        x2, y2 = w, int(calc_y(w))

    cv2.line(img, (x1, y1), (x2, y2), (0, 0, 255), 2)


for rho, theta in lines:
    draw_line(img, theta, rho)

    print(theta)
imshow(img)

Comments

[temps1101]

ありがとうございます。

[fana]

（OpenCV4.0.0のリファレンスを見たところ）
HoughLinesの引数にθの範囲の最大と最小があるようです．
この引数で最初から制限できるのではないかと見えます．

[temps1101]

ありがとうございます。参考にさせていただきます