skimage_sobel_edge_detection_with_matplotlib_visualization.py

1import matplotlib.pyplot as plt
2from skimage import data, filters
3
4# Load a sample image from the library
5image = data.coins()
6
7# Apply a Sobel filter to find edges
8edges = filters.sobel(image)
9
10# Create a figure to display the original and the processed image
11fig, axes = plt.subplots(ncols=2, figsize=(8, 4))
12ax = axes.ravel()
13
14ax[0].imshow(image, cmap=plt.cm.gray)
15ax[0].set_title('Original image')
16
17ax[1].imshow(edges, cmap=plt.cm.gray)
18ax[1].set_title('Sobel edges')
19
20for a in ax:
21    a.axis('off')
22
23plt.tight_layout()
24plt.show()