mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-29 00:23:43 +08:00
emmanuelle
112 lines
3.4 KiB
Python
112 lines
3.4 KiB
Python
import numpy as np
|
|
from skimage.segmentation import random_walker
|
|
try:
|
|
import pyamg
|
|
amg_loaded = True
|
|
except ImportError:
|
|
amg_loaded = False
|
|
|
|
|
|
def make_2d_syntheticdata(lx, ly=None):
|
|
if ly is None:
|
|
ly = lx
|
|
data = np.zeros((lx, ly)) + 0.1 * np.random.randn(lx, ly)
|
|
small_l = int(lx / 5)
|
|
data[lx / 2 - small_l:lx / 2 + small_l,
|
|
ly / 2 - small_l:ly / 2 + small_l] = 1
|
|
data[lx / 2 - small_l + 1:lx / 2 + small_l - 1,
|
|
ly / 2 - small_l + 1:ly / 2 + small_l - 1] = \
|
|
0.1 * np.random.randn(2 * small_l - 2, 2 * small_l - 2)
|
|
data[lx / 2 - small_l, ly / 2 - small_l / 8:ly / 2 + small_l / 8] = 0
|
|
seeds = np.zeros_like(data)
|
|
seeds[lx / 5, ly / 5] = 1
|
|
seeds[lx / 2 + small_l / 4, ly / 2 - small_l / 4] = 2
|
|
return data, seeds
|
|
|
|
|
|
def make_3d_syntheticdata(lx, ly=None, lz=None):
|
|
if ly is None:
|
|
ly = lx
|
|
if lz is None:
|
|
lz = lx
|
|
data = np.zeros((lx, ly, lz)) + 0.1 * np.random.randn(lx, ly, lz)
|
|
small_l = int(lx / 5)
|
|
data[lx / 2 - small_l:lx / 2 + small_l,
|
|
ly / 2 - small_l:ly / 2 + small_l,
|
|
lz / 2 - small_l:lz / 2 + small_l] = 1
|
|
data[lx / 2 - small_l + 1:lx / 2 + small_l - 1,
|
|
ly / 2 - small_l + 1:ly / 2 + small_l - 1,
|
|
lz / 2 - small_l + 1:lz / 2 + small_l - 1] = 0
|
|
# make a hole
|
|
hole_size = np.max([1, small_l / 8])
|
|
data[lx / 2 - small_l,
|
|
ly / 2 - hole_size:ly / 2 + hole_size,\
|
|
lz / 2 - hole_size:lz / 2 + hole_size] = 0
|
|
seeds = np.zeros_like(data)
|
|
seeds[lx / 5, ly / 5, lz / 5] = 1
|
|
seeds[lx / 2 + small_l / 4, ly / 2 - small_l / 4, lz / 2 - small_l / 4] = 2
|
|
return data, seeds
|
|
|
|
|
|
def test_2d_bf():
|
|
lx = 70
|
|
ly = 100
|
|
data, labels = make_2d_syntheticdata(lx, ly)
|
|
labels_bf = random_walker(data, labels, beta=90, mode='bf')
|
|
assert (labels_bf[25:45, 40:60] == 2).all()
|
|
return data, labels_bf
|
|
|
|
|
|
def test_2d_cg():
|
|
lx = 70
|
|
ly = 100
|
|
data, labels = make_2d_syntheticdata(lx, ly)
|
|
labels_cg = random_walker(data, labels, beta=90, mode='cg')
|
|
assert (labels_cg[25:45, 40:60] == 2).all()
|
|
return data, labels_cg
|
|
|
|
|
|
def test_2d_cg_mg():
|
|
lx = 70
|
|
ly = 100
|
|
data, labels = make_2d_syntheticdata(lx, ly)
|
|
labels_cg_mg = random_walker(data, labels, beta=90, mode='cg_mg')
|
|
assert (labels_cg_mg[25:45, 40:60] == 2).all()
|
|
return data, labels_cg_mg
|
|
|
|
|
|
def test_2d_inactive():
|
|
lx = 70
|
|
ly = 100
|
|
data, labels = make_2d_syntheticdata(lx, ly)
|
|
labels[10:20, 10:20] = -1
|
|
labels[46:50, 33:38] = -2
|
|
labels = random_walker(data, labels, beta=90)
|
|
assert (labels.reshape((lx, ly))[25:45, 40:60] == 2).all()
|
|
return data, labels
|
|
|
|
|
|
def test_3d():
|
|
n = 30
|
|
lx, ly, lz = n, n, n
|
|
data, labels = make_3d_syntheticdata(lx, ly, lz)
|
|
labels = random_walker(data, labels, mode='cg')
|
|
assert (labels.reshape(data.shape)[13:17, 13:17, 13:17] == 2).all()
|
|
return data, labels
|
|
|
|
|
|
def test_3d_inactive():
|
|
n = 30
|
|
lx, ly, lz = n, n, n
|
|
data, labels = make_3d_syntheticdata(lx, ly, lz)
|
|
old_labels = np.copy(labels)
|
|
labels[5:25, 26:29, 26:29] = -1
|
|
after_labels = np.copy(labels)
|
|
labels = random_walker(data, labels, mode='cg')
|
|
assert (labels.reshape(data.shape)[13:17, 13:17, 13:17] == 2).all()
|
|
return data, labels, old_labels, after_labels
|
|
|
|
if __name__ == '__main__':
|
|
from numpy import testing
|
|
testing.run_module_suite()
|