def test_random_brigtness(alpha, expected): img = np.ones((100, 100, 3), dtype=np.uint8) * 127 img = F.random_brightness(img, alpha) assert img.dtype == np.dtype('uint8') assert (img == expected).all()
def test_random_brightness_float(alpha, expected): img = np.ones((100, 100, 3), dtype=np.float32) * 0.4 expected = np.ones((100, 100, 3), dtype=np.float32) * expected img = F.random_brightness(img, alpha) assert img.dtype == np.dtype('float32') assert_array_almost_equal_nulp(img, expected)
def albumentations(self, img): return albumentations.random_brightness(img, alpha=1.5)
def test_random_brightness(alpha, expected): img = np.ones((100, 100, 3), dtype=np.uint8) * 127 img = F.random_brightness(img, alpha) assert img.dtype == np.dtype('uint8') assert (img == expected).all()
def albumentations(self, img): return albumentations.random_brightness(img, alpha=1.5)