mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-02 01:20:33 +08:00
Vighnesh Birodkar
30 lines
607 B
Python
30 lines
607 B
Python
import numpy as np
|
|
from skimage import graph
|
|
|
|
def test_threshold_cut():
|
|
|
|
img = np.zeros((100,100,3),dtype='uint8')
|
|
img[:50,:50] = 255,255,255
|
|
img[:50,50:] = 254,254,254
|
|
img[50:,:50] = 2,2,2
|
|
img[50:,50:] = 1,1,1
|
|
|
|
|
|
|
|
labels = np.zeros((100,100),dtype='uint8')
|
|
labels[:50,:50] = 0
|
|
labels[:50,50:] = 1
|
|
labels[50:,:50] = 2
|
|
labels[50:,50:] = 3
|
|
|
|
|
|
#print labels
|
|
rag = graph.rag_meancolor(img, labels)
|
|
#print "no of edges",rag.number_of_edges()
|
|
new_labels = graph.threshold_cut(labels, rag, 10)
|
|
|
|
assert new_labels.max() == 2
|
|
#assert False
|
|
|
|
|