def test___init___deterministic_settinga(self):
colorizer = iaa.RandomColorsBinaryImageColorizer(color_true=1,
color_false=2)
assert is_parameter_instance(colorizer.color_true, iap.Deterministic)
assert is_parameter_instance(colorizer.color_false, iap.Deterministic)
assert colorizer.color_true.value == 1
assert colorizer.color_false.value == 2
def test___init___default_settings(self):
colorizer = iaa.RandomColorsBinaryImageColorizer()
assert is_parameter_instance(colorizer.color_true, iap.DiscreteUniform)
assert is_parameter_instance(colorizer.color_false, iap.DiscreteUniform)
assert colorizer.color_true.a.value == 0
assert colorizer.color_true.b.value == 255
assert colorizer.color_false.a.value == 0
assert colorizer.color_false.b.value == 255
def test___init___custom_settings(self):
aug = iaa.AveragePooling(((2, 4), (5, 6)), keep_size=False)
assert len(aug.kernel_size) == 2
assert is_parameter_instance(aug.kernel_size[0], iap.DiscreteUniform)
assert is_parameter_instance(aug.kernel_size[1], iap.DiscreteUniform)
assert aug.kernel_size[0].a.value == 2
assert aug.kernel_size[0].b.value == 4
assert aug.kernel_size[1].a.value == 5
assert aug.kernel_size[1].b.value == 6
assert aug.keep_size is False
def test___init___tuple_and_list(self):
colorizer = iaa.RandomColorsBinaryImageColorizer(
color_true=(0, 100), color_false=[200, 201, 202])
assert is_parameter_instance(colorizer.color_true, iap.DiscreteUniform)
assert is_parameter_instance(colorizer.color_false, iap.Choice)
assert colorizer.color_true.a.value == 0
assert colorizer.color_true.b.value == 100
assert colorizer.color_false.a[0] == 200
assert colorizer.color_false.a[1] == 201
assert colorizer.color_false.a[2] == 202
def test___init__(self):
# check parameters
aug = iaa.FastSnowyLandscape(lightness_threshold=[100, 200],
lightness_multiplier=[1.0, 4.0])
assert is_parameter_instance(aug.lightness_threshold, iap.Choice)
assert len(aug.lightness_threshold.a) == 2
assert aug.lightness_threshold.a[0] == 100
assert aug.lightness_threshold.a[1] == 200
assert is_parameter_instance(aug.lightness_multiplier, iap.Choice)
assert len(aug.lightness_multiplier.a) == 2
assert np.allclose(aug.lightness_multiplier.a[0], 1.0)
assert np.allclose(aug.lightness_multiplier.a[1], 4.0)
def test___init___default_settings(self):
aug = iaa.AveragePooling(2)
assert len(aug.kernel_size) == 2
assert is_parameter_instance(aug.kernel_size[0], iap.Deterministic)
assert aug.kernel_size[0].value == 2
assert aug.kernel_size[1] is None
assert aug.keep_size is True
def test_get_parameters(self):
aug = self.augmenter(2)
params = aug.get_parameters()
assert len(params) == 2
assert len(params[0]) == 2
assert is_parameter_instance(params[0][0], iap.Deterministic)
assert params[0][0].value == 2
assert params[0][1] is None
def test___init___custom_settings(self):
aug = iaa.Canny(
alpha=0.2,
hysteresis_thresholds=([0, 1, 2], iap.DiscreteUniform(1, 10)),
sobel_kernel_size=[3, 5],
colorizer=iaa.RandomColorsBinaryImageColorizer(
color_true=10, color_false=20)
)
assert is_parameter_instance(aug.alpha, iap.Deterministic)
assert isinstance(aug.hysteresis_thresholds, tuple)
assert is_parameter_instance(aug.sobel_kernel_size, iap.Choice)
assert isinstance(aug.colorizer, iaa.RandomColorsBinaryImageColorizer)
assert np.isclose(aug.alpha.value, 0.2)
assert len(aug.hysteresis_thresholds) == 2
assert is_parameter_instance(aug.hysteresis_thresholds[0], iap.Choice)
assert aug.hysteresis_thresholds[0].a == [0, 1, 2]
assert is_parameter_instance(aug.hysteresis_thresholds[1],
iap.DiscreteUniform)
assert np.isclose(aug.hysteresis_thresholds[1].a.value, 1)
assert np.isclose(aug.hysteresis_thresholds[1].b.value, 10)
assert is_parameter_instance(aug.sobel_kernel_size, iap.Choice)
assert aug.sobel_kernel_size.a == [3, 5]
assert is_parameter_instance(aug.colorizer.color_true,
iap.Deterministic)
assert is_parameter_instance(aug.colorizer.color_false,
iap.Deterministic)
assert aug.colorizer.color_true.value == 10
assert aug.colorizer.color_false.value == 20
def test___init___default_settings(self):
aug = iaa.Canny()
assert is_parameter_instance(aug.alpha, iap.Uniform)
assert isinstance(aug.hysteresis_thresholds, tuple)
assert is_parameter_instance(aug.sobel_kernel_size, iap.DiscreteUniform)
assert isinstance(aug.colorizer, iaa.RandomColorsBinaryImageColorizer)
assert np.isclose(aug.alpha.a.value, 0.0)
assert np.isclose(aug.alpha.b.value, 1.0)
assert len(aug.hysteresis_thresholds) == 2
assert is_parameter_instance(aug.hysteresis_thresholds[0],
iap.DiscreteUniform)
assert np.isclose(aug.hysteresis_thresholds[0].a.value, 100-40)
assert np.isclose(aug.hysteresis_thresholds[0].b.value, 100+40)
assert is_parameter_instance(aug.hysteresis_thresholds[1],
iap.DiscreteUniform)
assert np.isclose(aug.hysteresis_thresholds[1].a.value, 200-40)
assert np.isclose(aug.hysteresis_thresholds[1].b.value, 200+40)
assert aug.sobel_kernel_size.a.value == 3
assert aug.sobel_kernel_size.b.value == 7
assert is_parameter_instance(aug.colorizer.color_true,
iap.DiscreteUniform)
assert is_parameter_instance(aug.colorizer.color_false,
iap.DiscreteUniform)
assert aug.colorizer.color_true.a.value == 0
assert aug.colorizer.color_true.b.value == 255
assert aug.colorizer.color_false.a.value == 0
assert aug.colorizer.color_false.b.value == 255
def test___init___single_value_hysteresis(self):
aug = iaa.Canny(
alpha=0.2,
hysteresis_thresholds=[0, 1, 2],
sobel_kernel_size=[3, 5],
colorizer=iaa.RandomColorsBinaryImageColorizer(
color_true=10, color_false=20)
)
assert is_parameter_instance(aug.alpha, iap.Deterministic)
assert is_parameter_instance(aug.hysteresis_thresholds, iap.Choice)
assert is_parameter_instance(aug.sobel_kernel_size, iap.Choice)
assert isinstance(aug.colorizer, iaa.RandomColorsBinaryImageColorizer)
assert np.isclose(aug.alpha.value, 0.2)
assert aug.hysteresis_thresholds.a == [0, 1, 2]
assert is_parameter_instance(aug.sobel_kernel_size, iap.Choice)
assert aug.sobel_kernel_size.a == [3, 5]
assert is_parameter_instance(aug.colorizer.color_true,
iap.Deterministic)
assert is_parameter_instance(aug.colorizer.color_false,
iap.Deterministic)
assert aug.colorizer.color_true.value == 10
assert aug.colorizer.color_false.value == 20
def test_get_parameters(self):
aug = self.create_aug(p=0.5)
params = aug.get_parameters()
assert is_parameter_instance(params[0], iap.Binomial)
assert is_parameter_instance(params[0].p, iap.Deterministic)
assert 0.5 - 1e-4 < params[0].p.value < 0.5 + 1e-4
