mirror of
https://github.com/wassname/scikit-image.git
synced 2026-08-12 12:30:16 +08:00
Improved drawing in plot_rag_draw.py example (#1872)
Improved drawing in draw_rag example; changed draw_rag to show_rag, made it return ScalarMappable to draw colorbars; added magma colormap
This commit is contained in:
committed by
Egor Panfilov
parent
e6fd683d74
commit
a406e270b1
@@ -8,7 +8,7 @@ This example demonstrates construction of region boundary based RAGs with the
|
||||
"""
|
||||
from skimage.future import graph
|
||||
from skimage import data, segmentation, color, filters, io
|
||||
from skimage.util.colormap import viridis
|
||||
from matplotlib import pyplot as plt
|
||||
|
||||
|
||||
img = data.coffee()
|
||||
@@ -19,9 +19,8 @@ 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)
|
||||
|
||||
out = graph.draw_rag(labels, g, edges_rgb, node_color="#999999",
|
||||
colormap=viridis)
|
||||
|
||||
io.imshow(out)
|
||||
plt.colorbar(lc, fraction=0.03)
|
||||
io.show()
|
||||
|
||||
Reference in New Issue
Block a user