Function get_data

dali/test/python/test_dali_variable_batch_size.py:784–793

(batch_size)

        return pipe

    def get_data(batch_size):
        test_data_shape = [random.randint(5, 21), random.randint(5, 21), 3]
        data1 = [
            np.random.randint(0, 255, size=test_data_shape, dtype=np.uint8)
            for _ in range(batch_size)
        ]
        data2 = [
            np.random.randint(1, 4, size=test_data_shape, dtype=np.uint8) for _ in range(batch_size)
        ]
        return (data1, data2)

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