def test_all_constraints_3d(self):
test_x, test_y, test_z = 1, 2, 3
aff_c = AffineWarpConstraints.translation_3d(
x=test_x, y=test_y, z=test_z)
self.assertEqual(
aff_c.constraints,
((None, None, None, 1),
(None, None, None, 2),
(None, None, None, 3)))
aff_c = AffineWarpConstraints.scale_3d(
x=test_x, y=test_y, z=test_z)
self.assertEqual(
aff_c.constraints,
((1, None, None, None),
(None, 2, None, None),
(None, None, 3, None)))
aff_c = AffineWarpConstraints.no_shear_3d()
self.assertEqual(
aff_c.constraints,
((None, 0, 0, None),
(0, None, 0, None),
(0, 0, None, None)))
def test_default(self):
aff_c = AffineWarpConstraints()
self.assertAllEqual(aff_c.constraints,
((None, None, None), (None, None, None)))
self.assertAllClose(aff_c.num_free_params, 6)
self.assertAllClose(aff_c.num_dim, 2)
customise_constraints = ((None, None, 1, 1), (None, None, None, 2),
(None, 1, None, 3))
aff_c = AffineWarpConstraints(customise_constraints)
self.assertEqual(aff_c.constraints, customise_constraints)
def test_all_constraints_2d(self):
test_x, test_y = 1, 2
aff_c = AffineWarpConstraints.translation_2d(x=test_x, y=test_y)
self.assertEqual(aff_c.constraints, ((None, None, 1), (None, None, 2)))
aff_c = AffineWarpConstraints.scale_2d(x=test_x, y=test_y)
self.assertEqual(aff_c.constraints, ((1, None, None), (None, 2, None)))
aff_c = AffineWarpConstraints.shear_2d(x=test_x, y=test_y)
self.assertEqual(aff_c.constraints, ((None, 1, None), (2, None, None)))
aff_c = AffineWarpConstraints.no_shear_2d()
self.assertEqual(aff_c.constraints, ((None, 0, None), (0, None, None)))
def test_constraints(self):
grid_warper = AffineGridWarperLayer(
source_shape=(3, 3),
output_shape=(2, 4),
constraints=AffineWarpConstraints.no_shear_2d())
self.assertEqual(grid_warper.constraints.constraints,
((None, 0, None), (0, None, None)))
def test_3d_3d_translation(self):
aff = tf.constant([[2, 0, 0,
0, 1.5, 0,
0, 0, 3]], dtype=tf.float32)
params = {'source_shape': (3, 3, 3), 'output_shape': (2, 2),
'constraints': AffineWarpConstraints.translation_3d(3, 4, 5)}
expected = [[[[3, 4, 5], [3, 5.5, 5]], [[5, 4, 5], [5, 5.5, 5]]]]
self._test_correctness(params, aff, expected)
def test_3d_2d_scale(self):
aff = tf.constant([[0, 0, 3,
0, 0, 4,
0, 0, 5]], dtype=tf.float32)
params = {'source_shape': (3, 3, 3), 'output_shape': (2, 2),
'constraints': AffineWarpConstraints.scale_3d(1, 1, 1)}
expected = [[[[3, 4, 5], [3, 5, 5]], [[4, 4, 5], [4, 5, 5]]]]
self._test_correctness(params, aff, expected)
def test_combine_constraints(self):
test_x, test_y, test_z = 1, 2, 3
aff_c_1 = AffineWarpConstraints.translation_2d(x=test_x, y=test_y)
aff_c_2 = AffineWarpConstraints.shear_2d(x=test_x, y=test_y)
aff_comb = aff_c_1.combine_with(aff_c_2)
self.assertEqual(aff_comb.constraints, ((None, 1, 1), (2, None, 2)))
aff_c_1 = AffineWarpConstraints.translation_3d(
x=test_x, y=test_y, z=test_z)
aff_c_2 = AffineWarpConstraints.no_shear_3d()
aff_comb = aff_c_1.combine_with(aff_c_2)
self.assertEqual(aff_comb.constraints,
((None, 0, 0, 1), (0, None, 0, 2), (0, 0, None, 3)))
aff_c_1 = AffineWarpConstraints.translation_2d(x=test_x, y=test_y)
aff_c_2 = AffineWarpConstraints.no_shear_3d()
with self.assertRaisesRegexp(ValueError, ''):
aff_c_1.combine_with(aff_c_2)
def test_no_constraints(self):
aff_c = AffineWarpConstraints.no_constraints(num_dim=2)
self.assertEqual(aff_c.constraints,
((None, None, None), (None, None, None)))