Function test_remap

dali/test/python/test_dali_variable_batch_size.py:846–872

()

def test_remap():
    def pipe(max_batch_size, input_data, device):
        pipe = Pipeline(batch_size=max_batch_size, device_id=0, num_threads=4)
        with pipe:
            input, mapx, mapy = fn.external_source(device=device, source=input_data, num_outputs=3)
            out = fn.experimental.remap(input, mapx, mapy)
        pipe.set_outputs(out)
        return pipe

    def get_data(batch_size):
        input_shape = [480, 640, 3]
        mapx_shape = mapy_shape = [480, 640]
        input = [
            np.random.randint(0, 255, size=input_shape, dtype=np.uint8) for _ in range(batch_size)
        ]
        mapx = [
            640 * np.random.random(size=mapx_shape).astype(np.float32)  # [0, 640) interval
            for _ in range(batch_size)
        ]
        mapy = [
            480 * np.random.random(size=mapy_shape).astype(np.float32)  # [0, 480) interval
            for _ in range(batch_size)
        ]
        return input, mapx, mapy

    input_data = [get_data(random.randint(5, 31)) for _ in range(13)]
    check_pipeline(input_data, pipeline_fn=pipe, devices=["gpu"])


def test_random_bbox_crop_op():