Hough変換法による直線識別について

Question

前提・実現したいこと

Python初心者です．
Hough変換法を用いて以下のような画像から直線を識別しようと試みています．

（Hough変換のソースコードについてはhttp://labs.eecs.tottori-u.ac.jp/sd/Member/oyamada/OpenCV/html/py_tutorials/py_imgproc/py_houghlines/py_houghlines.htmlを参考にさせていただきました．）

発生している問題・エラーメッセージ

上図のように画像中には複数の線が含まれているのですが，実装してみると以下のように一つの直線しか認識されていません．

該当のソースコード

import cv2
import numpy as np

img = cv2.imread('image.jpg')
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
edges = cv2.Canny(gray,50,150,apertureSize = 3)

lines = cv2.HoughLines(edges,1,np.pi/180,50)
for rho,theta in lines[0]:
    a = np.cos(theta)
    b = np.sin(theta)
    x0 = a*rho
    y0 = b*rho
    x1 = int(x0 + 1000*(-b))
    y1 = int(y0 + 1000*(a))
    x2 = int(x0 - 1000*(-b))
    y2 = int(y0 - 1000*(a))

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

cv2.imwrite('houghlines.jpg',img)

試したこと

関数HoughLinesの引数を調整し，いくつか実行してみたのですが，いまいちうまくいきません．

補足情報（FW/ツールのバージョンなど）

Python初心者ですので，なるべく詳しく解説していただけると幸いです．

Answer 1

下記の部分ですが、複数の候補が含まれるlinesから1つだけ(lines[0]だけ)取り出して直線を描いています。

for rho,theta in lines[0]:
    a = np.cos(theta)
    ...

この部分を複数の候補を全て描画するように、下記のように変更する必要があります(インデントもすべて直す必要があります)。

for line in lines:
    for rho,theta in line:
        a = np.cos(theta)
        ...

またこの例では最終的にハフ変換した画像だけ保存していますが、白黒化したgrayやエッジ抽出したedgesも同様に保存すると何故意図した結果にならないのか、分かりやすいと思います。一番最後の行に次の2行を追加して各々画像を確認してみて下さい。

cv2.imwrite('gray.jpg',gray)
cv2.imwrite('edges.jpg',edges)

Comments

[monkey.0435]

ご回答ありがとうございます！
linesの部分を見落としていました...

いろいろ試してみます！