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,