Function test_resize_alignment

dali/test/python/operator_2/test_tensor_resize.py:175–205

(device)

@params("cpu", "gpu")
def test_resize_alignment(device):
    data = data_1x1x2x4
    scales = np.array([1.0, 1.0, 0.6, 0.6], dtype=np.float32)

    def resize_fn(input_data, alignment=0):
        return fn.tensor_resize(
            input_data,
            scales=scales,
            alignment=alignment,
            interp_type=types.INTERP_LINEAR,
            antialias=False,
        )

    run_and_compare(
        np.array([[[[2.6666665, 4.3333331]]]], dtype=np.float32),
        data,
        device,
        lambda in_data: resize_fn(in_data, alignment=0),
    )
    run_and_compare(
        np.array([[[[3.6666665, 5.3333331]]]], dtype=np.float32),
        data,
        device,
        lambda in_data: resize_fn(in_data, alignment=0.5),
    )
    run_and_compare(
        np.array([[[[4.6666665, 6.3333331]]]], dtype=np.float32),
        data,
        device,
        lambda in_data: resize_fn(in_data, alignment=1),
    )


@params("cpu", "gpu")