博士のつぶやき

OpenCVを活用した画像認識のテスト。

今日は、ハフ変換を実施。

import cv2
import numpy as np

img = cv2.imread('**.jpg')
img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
img_canny = cv2.Canny(img_gray, 50, 150, apertureSize = 3)

lines = cv2.HoughLines(img, 1, np.pi/180, 200)

for line in lines:
    rho, theta = line[0]

で、原点からのラインの距離と角度を読みだす。
ラインを引きたいときは、
cv2.line(img, ・・・）を使って書き込むと、どの直線が抽出されたか明らかになる。

block_hough.pyに実装した。