def segmentation(string):
if string:
mask = parse_jsonstring(string, img.shape)
seg = watershed_segmentation(img, mask)
src = color.label2rgb(seg, image=img)
else:
raise PreventUpdate
return array_to_data_url(img_as_ubyte(src))
def segmentation(string):
if string:
mask = parse_jsonstring(string,
io.imread(filename, as_gray=True).shape)
seg = watershed_segmentation(io.imread(filename, as_gray=True), mask)
src = color.label2rgb(seg, image=io.imread(filename, as_gray=True))
else:
raise PreventUpdate
return array_to_data_url(img_as_ubyte(src))
def test_watershed_segmentation():
img = np.zeros((20, 20))
img[2:6, 2:6] = 1
img[10:15, 10:15] = 1
mask = np.zeros_like(img, dtype=np.uint8)
mask[4, 4] = 1
mask[12, 12] = 2
res = watershed_segmentation(img, mask, sigma=0.1)
assert np.all(res[2:6, 2:6] == 1)
assert np.all(res[10:15, 10:15] == 2)