ハフ変換の直線検出がうまくいきません

Question

ハフ変換による直線検出を勉強しています。
ここのサイトを参考に、同じ画像で同じコードを使ってもサイトのような直線検出を表示できませんでした。

どのようにすれば検出できるのでしょうか？

import cv2
import numpy as np

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

lines = cv2.HoughLines(edges,1,np.pi/180,200)
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('houghlines3.jpg',img)

Comments

[can110]

元画像「dave.jpg」の取得先（リンクURLなど）を提示ください。

Answer 1

ハフ変換による直線検出で示されたソースコードでは、検出された最初の１直線しか描画されないようです。
検出された全直線を描画するようなソース

import cv2
import numpy as np

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

lines = cv2.HoughLines(edges,1,np.pi/180,200)
print(len(lines)) # 20本
for line in lines:
    for rho,theta in line:
        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('houghlines3.jpg',img)

にてWhere can I locate the demo image dave.jpg? [closed]にあげられた画像を利用した結果は以下のようになりました。

Comments

[syayo]

ありがとうございました。
解決しました。
教えてくださったコードに
linesのパラメータを調整したところうまくいきました。

[can110]

おそらく元画像のサイズが小さかったのだと思われます。
その場合は、HoughLines関数の第4引数＝検出可能な線の長さの最小値も小さい値を与える必要があります。

[syayo]

おっしゃる通り、画像のサイズが小さかったです。

ありがとうございます。
パラメータの意味もしっかり学びます。