>>> import sys
>>> import os
>>> import math
>>> import numpy as np
>>> from scipy import ndimage as ndi
>>> from skimage.transform import hough_line, hough_line_peaks
>>> from scipy.ndimage import gaussian_filter
>>> from PIL import Image
>>> from matplotlib import pyplot as plt
>>> def get_size( my_image ):
img = Image.open(my_image)
x,y = img.size
return (x,y)
>>> def create_grayscale( my_image ):
grayscale = np.array(Image.open( my_image ).convert('L'))
return grayscale
>>> def show_grayscale( my_image ):
plt.figure(figsize = (10,10))
plt.imshow(my_image, cmap = plt.get_cmap(name = 'gray'))
plt.show()
>>> def LoG (my_image, sigma):
A = gaussian_filter(grayscale, sigma=math.sqrt(sigma)/math.sqrt(2))
B = gaussian_filter(grayscale, sigma=math.sqrt(sigma)*math.sqrt(2))
return np.subtract(A,B)
>>> def Gradient( pixel_arr ):
padded_arr = np.pad(pixel_arr,1)
Sx_prewitt = [[-1,-1,-1],[0,0,0],[1,1,1]]
Sy_prewitt = [[-1,0,1],[-1,0,1],[-1,0,1]]
Gx = []
Gy = []
for i in range(len(pixel_arr)):
row_x = []
row_y = []
for j in range(len(pixel_arr[0])):
neigh_matrix = padded_arr[i:i+3,j:j+3]
g_x = np.multiply(neigh_matrix,Sx_prewitt)
>>> def Gradient( pixel_arr ):
padded_arr = np.pad(pixel_arr,1)
Sx_prewitt = [[-1,-1,-1],[0,0,0],[1,1,1]]
Sy_prewitt = [[-1,0,1],[-1,0,1],[-1,0,1]]
Gx = []
Gy = []
for i in range(len(pixel_arr)):
row_x = []
row_y = []
for j in range(len(pixel_arr[0])):
neigh_matrix = padded_arr[i:i+3,j:j+3]
g_x = np.multiply(neigh_matrix,Sx_prewitt)
g_y = np.multiply(neigh_matrix,Sy_prewitt)
sum_x = np.sum(g_x)
sum_y = np.sum(g_y)
row_x.append(sum_x)
row_y.append(sum_y)
Gx.append(row_x)
Gy.append(row_y)
Gx2 = np.multiply(Gx,Gx)
Gy2 = np.multiply(Gy,Gy)
sum_Gx2 = Gx2 + Gy2
gradient_mag = np.sqrt(sum_Gx2)
print(gradient_mag)
gradient_mag[gradient_mag < 60] = 0
edge_img = Image.fromarray(gradient_mag)
show_grayscale(edge_img)
return edge_img
>>> def generate_Hough_Transform(my_image):
h, theta, d = hough_line(my_image)
fig, axes = plt.subplots(1, 3, figsize=(15, 6))
ax = axes.ravel()
ax[0].imshow(my_image, cmap = plt.get_cmap(name = 'gray'))
ax[0].set_title('Input image')
ax[0].set_axis_off()
ax[1].imshow(np.log(1 + h),
extent=[np.rad2deg(theta[-1]), np.rad2deg(theta[0]), d[-1], d[0]],
cmap = plt.get_cmap(name = 'gray'), aspect=1/1.5)
ax[1].set_title('Hough transform')
ax[1].set_xlabel('Angles (degrees)')
ax[1].set_ylabel('Distance (pixels)')
ax[1].axis('on')
ax[2].imshow(my_image, cmap = plt.get_cmap(name = 'gray'))
origin = np.array((0, grayscale.shape[1]))
for _, angle, dist in zip(*hough_line_peaks(h, theta, d)):
y0, y1 = (dist - origin * np.cos(angle)) / np.sin(angle)
ax[2].plot(origin, (y0, y1), '-r')
ax[2].set_xlim(origin)
ax[2].set_ylim((my_image.shape[0], 0))
ax[2].set_axis_off()
ax[2].set_title('Detected lines')
plt.tight_layout()
plt.show()
>>> grayscale = create_grayscale('/Users/spurs/Desktop/IMG_2211.jpeg')
>>> show_grayscale(grayscale)
>>> show_grayscale(grayscale)
>>> (size_x, size_y) = get_size('/Users/spurs/Desktop/IMG_2211.jpeg')
>>> (size_x, size_y) = get_size('/Users/spurs/Desktop/myselfgray.png')
>>> LoG_Image = LoG( grayscale, sigma=5)
>>> generate_Hough_Transform(LoG_Image)
>>> grayscale = create_grayscale('/Users/spurs/Desktop/IMG_2211.jpeg')
>>> show_grayscale(grayscale)
>>> (size_x, size_y) = get_size('corner_window.jpg')
效果图: