mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-29 21:26:21 +08:00
Vighnesh Birodkar
27 lines
636 B
Python
27 lines
636 B
Python
import networkx as nx
|
|
import numpy as np
|
|
|
|
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)
|
|
|
|
rag.remove_edges_from(to_remove)
|
|
|
|
#print "to remove", to_remove
|
|
|
|
comps = nx.connected_components(rag)
|
|
out = np.copy(label)
|
|
#print "comps",len(comps)
|
|
|
|
for i, nodes in enumerate(comps) :
|
|
|
|
for node in nodes :
|
|
for l in rag.node[node]['labels'] :
|
|
out[label == l] = i
|
|
|
|
#print out
|
|
#print label
|
|
return out
|