mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-27 23:23:26 +08:00
Vighnesh Birodkar
a406e270b1
Improved drawing in draw_rag example; changed draw_rag to show_rag, made it return ScalarMappable to draw colorbars; added magma colormap
27 lines
694 B
Python
27 lines
694 B
Python
"""
|
|
==========================
|
|
Region Boundary based RAGs
|
|
==========================
|
|
|
|
This example demonstrates construction of region boundary based RAGs with the
|
|
`rag_boundary` function.
|
|
"""
|
|
from skimage.future import graph
|
|
from skimage import data, segmentation, color, filters, io
|
|
from matplotlib import pyplot as plt
|
|
|
|
|
|
img = data.coffee()
|
|
gimg = color.rgb2gray(img)
|
|
|
|
labels = segmentation.slic(img, compactness=30, n_segments=400)
|
|
edges = filters.sobel(gimg)
|
|
edges_rgb = color.gray2rgb(edges)
|
|
|
|
g = graph.rag_boundary(labels, edges)
|
|
lc = graph.show_rag(labels, g, edges_rgb, img_cmap=None, edge_cmap='viridis',
|
|
edge_width=1.2)
|
|
|
|
plt.colorbar(lc, fraction=0.03)
|
|
io.show()
|