def test_boundary_exception():
ATMBuilder(boundary=-1).build(training, templates[1])
filenames = ['breakingbad.jpg', 'takeo.ppm', 'lenna.png', 'einstein.jpg']
training_shapes = []
templates = []
for i in range(4):
im = mio.import_builtin_asset(filenames[i])
im.crop_to_landmarks_proportion_inplace(0.1)
if im.n_channels == 3:
im = im.as_greyscale(mode='luminosity')
training_shapes.append(im.landmarks[None].lms)
templates.append(im)
# build atm
atm1 = ATMBuilder(features=igo,
transform=DifferentiablePiecewiseAffine,
normalization_diagonal=150,
n_levels=3,
downscale=2,
scaled_shape_models=True,
max_shape_components=[1, 2, 3],
boundary=3).build(training_shapes, templates[0])
atm2 = ATMBuilder(features=igo,
transform=DifferentiablePiecewiseAffine,
normalization_diagonal=150,
n_levels=1,
downscale=2,
scaled_shape_models=True,
max_shape_components=[1],
boundary=3).build(training_shapes, templates[1])
atm3 = ATMBuilder(features=igo,
transform=DifferentiablePiecewiseAffine,
def test_normalization_diagonal_exception():
atm = ATMBuilder(normalization_diagonal=100).build(training, templates[3])
assert (atm.warped_templates[0].n_true_pixels() == 1246)
ATMBuilder(normalization_diagonal=10).build(training, templates[3])
def test_max_shape_components_exception_2():
ATMBuilder(max_shape_components=[1, 2]).build(training, templates[0])
def test_downscale_exception():
atm = ATMBuilder(downscale=1).build(training, templates[2])
assert (atm.downscale == 1)
ATMBuilder(downscale=0).build(training, templates[2])
def test_n_levels_exception():
ATMBuilder(n_levels=0).build(training, templates[1])
def test_features_exception():
ATMBuilder(features=[igo, sparse_hog]).build(training, templates[0])
def test_verbose_mock(mock_stdout):
ATMBuilder().build(training, templates[2], verbose=True)
# load images
filenames = ['breakingbad.jpg', 'takeo.ppm', 'lenna.png', 'einstein.jpg']
training = []
templates = []
for i in range(4):
im = mio.import_builtin_asset(filenames[i])
if im.n_channels == 3:
im = im.as_greyscale(mode='luminosity')
training.append(im.landmarks[None].lms)
templates.append(im)
# build atms
atm1 = ATMBuilder(features=[igo, sparse_hog, no_op],
transform=PiecewiseAffine,
normalization_diagonal=150,
n_levels=3,
downscale=2,
scaled_shape_models=False,
max_shape_components=[1, 2, 3],
boundary=3).build(training, templates[0])
atm2 = ATMBuilder(features=[no_op, no_op],
transform=ThinPlateSplines,
trilist=None,
normalization_diagonal=None,
n_levels=2,
downscale=1.2,
scaled_shape_models=True,
max_shape_components=None,
boundary=0).build(training, templates[1])
atm3 = ATMBuilder(features=igo,