Exemplo n.º 1
0
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)
Exemplo n.º 2
0
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)
Exemplo n.º 3
0
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)
Exemplo n.º 4
0
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)
Exemplo n.º 5
0
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])))
Exemplo n.º 6
0
def test_image_as_vector():
    pixels = np.random.rand(10, 20, 1)
    image = Image(pixels)
    assert(np.all(image.as_vector() == pixels.ravel()))
Exemplo n.º 7
0
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])))
Exemplo n.º 8
0
def test_image_as_vector():
    pixels = np.random.rand(10, 20, 1)
    image = Image(pixels)
    assert (np.all(image.as_vector() == pixels.ravel()))
Exemplo n.º 9
0
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]))
Exemplo n.º 10
0
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]))