def test_normalize_norm_image_per_channel(): pixels = np.random.randn(120, 120, 3) pixels[..., 1] *= 17 pixels[..., 0] += -114 pixels[..., 2] /= 30 image = Image(pixels) image.normalize_norm_inplace(mode='per_channel') assert_allclose( np.mean(image.as_vector(keep_channels=True), axis=0), 0, atol=1e-10) assert_allclose( np.linalg.norm(image.as_vector(keep_channels=True), axis=0), 1)
def test_normalize_std_image_per_channel(): pixels = np.random.randn(120, 120, 3) pixels[..., 1] *= 9 pixels[..., 0] += -3 pixels[..., 2] /= 140 image = Image(pixels) image.normalize_std_inplace(mode='per_channel') assert_allclose(np.mean(image.as_vector(keep_channels=True), axis=0), 0, atol=1e-10) assert_allclose(np.std(image.as_vector(keep_channels=True), axis=0), 1)
def test_normalize_std_image_per_channel(): pixels = np.random.randn(3, 120, 120) pixels[1] *= 9 pixels[0] += -3 pixels[2] /= 140 image = Image(pixels) image.normalize_std_inplace(mode='per_channel') assert_allclose( np.mean(image.as_vector(keep_channels=True), axis=1), 0, atol=1e-10) assert_allclose( np.std(image.as_vector(keep_channels=True), axis=1), 1)
def test_normalize_norm_image_per_channel(): pixels = np.random.randn(3, 120, 120) pixels[1] *= 17 pixels[0] += -114 pixels[2] /= 30 image = Image(pixels) image.normalize_norm_inplace(mode='per_channel') assert_allclose(np.mean(image.as_vector(keep_channels=True), axis=1), 0, atol=1e-10) assert_allclose( np.linalg.norm(image.as_vector(keep_channels=True), axis=1), 1)
def test_image_as_vector_keep_channels(): pixels = np.random.rand(10, 20, 2) image = Image(pixels) assert(np.all(image.as_vector(keep_channels=True) == pixels.reshape([-1, 2])))
def test_image_as_vector(): pixels = np.random.rand(10, 20, 1) image = Image(pixels) assert(np.all(image.as_vector() == pixels.ravel()))
def test_image_as_vector_keep_channels(): pixels = np.random.rand(10, 20, 2) image = Image(pixels) assert (np.all( image.as_vector(keep_channels=True) == pixels.reshape([-1, 2])))
def test_image_as_vector(): pixels = np.random.rand(10, 20, 1) image = Image(pixels) assert (np.all(image.as_vector() == pixels.ravel()))
def test_image_as_vector_keep_channels(): pixels = np.random.rand(2, 10, 20) image = Image(pixels) assert np.all( image.as_vector(keep_channels=True) == pixels.reshape([2, -1]))
def test_image_as_vector_keep_channels(): pixels = np.random.rand(2, 10, 20) image = Image(pixels) assert np.all(image.as_vector(keep_channels=True) == pixels.reshape([2, -1]))