wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-08-12 12:30:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added example for region boundary merging

Browse Source
This commit is contained in:
Vighnesh Birodkar
2016-04-27 01:03:09 -04:00
parent 2a87c07800
commit b3f6b545cb
4 changed files with 82 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/examples/segmentation/plot_rag_merge.py
+5 -4
View File
@@ -31,13 +31,14 @@ def _weight_mean_color(graph, src, dst, n):
Returns
-------
weight : float
The absolute difference of the mean color between node `dst` and `n`.
data : dict
A dictionary with the `"weight"` attribute set as the absolute
difference of the mean color between node `dst` and `n`.
"""
diff = graph.node[dst]['mean color'] - graph.node[n]['mean color']
diff = np.linalg.norm(diff)
return diff
return {'weight': diff}
def merge_mean_color(graph, src, dst):
@@ -62,7 +63,7 @@ img = data.coffee()
labels = segmentation.slic(img, compactness=30, n_segments=400)
g = graph.rag_mean_color(img, labels)
labels2 = graph.merge_hierarchical(labels, g, thresh=40, rag_copy=False,
labels2 = graph.merge_hierarchical(labels, g, thresh=35, rag_copy=False,
in_place_merge=True,
merge_func=merge_mean_color,
weight_func=_weight_mean_color)