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