Files
scikit-image/doc/examples/plot_rag_merge.py
T
Vighnesh Birodkar b35fe6948f working code
2015-01-27 21:33:21 +05:30

14 lines
405 B
Python

from skimage import graph, data, io, segmentation, color
img = data.coffee()
labels = segmentation.slic(img, compactness=30, n_segments=400)
g = graph.rag_mean_color(img, labels)
labels2 = graph.merge_hierarchical(g, labels, 40)
g2 = graph.rag_mean_color(img, labels2)
out = color.label2rgb(labels2, img, kind='avg')
out = segmentation.mark_boundaries(out, labels2, (0, 0, 0))
io.imshow(out)
io.show()