mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-29 16:16:07 +08:00
emmanuelle
31 lines
659 B
Python
31 lines
659 B
Python
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
|
|
from skimage.transform import hough
|
|
|
|
img = np.zeros((100, 150), dtype=bool)
|
|
img[30, :] = 1
|
|
img[:, 65] = 1
|
|
img[35:45, 35:50] = 1
|
|
for i in range(90):
|
|
img[i, i] = 1
|
|
img += np.random.random(img.shape) > 0.95
|
|
|
|
out, angles, d = hough(img)
|
|
|
|
plt.subplot(1, 2, 1)
|
|
|
|
plt.imshow(img, cmap=plt.cm.gray)
|
|
plt.title('Input image')
|
|
|
|
plt.subplot(1, 2, 2)
|
|
plt.imshow(out, cmap=plt.cm.bone,
|
|
extent=(d[0], d[-1],
|
|
np.rad2deg(angles[0]), np.rad2deg(angles[-1])))
|
|
plt.title('Hough transform')
|
|
plt.xlabel('Angle (degree)')
|
|
plt.ylabel('Distance (pixel)')
|
|
|
|
plt.subplots_adjust(wspace=0.4)
|
|
plt.show()
|