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

Added testcase

Browse Source
This commit is contained in:
Vighnesh Birodkar
2014-06-19 02:50:24 +05:30
parent a6c9a5a2a7
commit abcb9cf2ef
5 changed files with 34 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/graph/graph_cut.py
+3 -2
View File
@@ -5,14 +5,15 @@ def threshold_cut(label, rag, thresh):
#print [rag.edges_iter(data = True)]
to_remove = [(x,y) for x,y,d in rag.edges_iter(data = True) if d['weight'] >= thresh]
print "edges to remove",len(to_remove)
#print "edges to remove",len(to_remove)
rag.remove_edges_from(to_remove)
#print "to remove", to_remove
comps = nx.connected_components(rag)
out = np.copy(label)
print "comps",len(comps)
#print "comps",len(comps)
for i, nodes in enumerate(comps) :