Docstrings

This commit is contained in:
Vighnesh Birodkar
2014-08-15 14:10:09 +05:30
parent f4aa0fc8e1
commit 5c662b4472
3 changed files with 67 additions and 13 deletions
+11 -5
View File
@@ -1,22 +1,28 @@
"""
===========
RAG Drawing
===========
This example constructs a Region Adjacency Graph (RAG) and draws it with
the `rag_draw` method.
"""
from skimage import graph, data, segmentation
from matplotlib import pyplot as plt
img = data.coffee()
labels = segmentation.slic(img, compactness=30, n_segments=400)
g = graph.rag_mean_color(img, labels)
out = graph.rag.rag_draw(labels, g, img)
out = graph.rag_draw(labels, g, img)
plt.figure()
plt.title("RAG with all edges shown in green.")
plt.imshow(out)
out = graph.rag.rag_draw(labels, g, img, high_color=(1, 0, 0), thresh=30)
out = graph.rag_draw(labels, g, img, high_color=(1, 0, 0), thresh=30)
plt.figure()
plt.title("RAG with edge weights less than 30,\
color mapped between green and red.")
plt.imshow(out)
plt.imshow(out)
plt.show()