-
Notifications
You must be signed in to change notification settings - Fork 0
/
26_hough_line.py
28 lines (25 loc) · 947 Bytes
/
26_hough_line.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import cv2 as cv
import numpy as np
img = cv.imread('images/sudoku.png')
grayImg = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
edges = cv.Canny(grayImg, 50, 150, apertureSize=3)
cv.imshow('Edges', edges)
lines = cv.HoughLines(edges, 1, np.pi / 180, 200)
for line in lines:
rho, theta = line[0]
a = np.cos(theta)
b = np.sin(theta)
x0 = a * rho
y0 = b * rho
# x1 stores the rounded off value of (r * cos(theta) - 1000 * sin(theta))
x1 = int(x0 + 1000 * (-b))
# y1 stores the rounded off value of (r * sin(theta) + 1000 * cos(theta))
y1 = int(y0 + 1000 * (a))
# x2 stores the rounded off value of (r * cos(theta) + 1000 * sin(theta))
x2 = int(x0 - 1000 * (-b))
# y1 stores the rounded off value of (r * sin(theta) - 1000 * cos(theta))
y2 = int(y0 - 1000 * (a))
cv.line(img, (x1, y1), (x2, y2), (0, 0, 255), 1)
cv.imshow('Image', img)
cv.waitKey(0)
cv.destroyAllWindows()