def pipe(max_batch_size, input_data, device):
pipe = Pipeline(batch_size=max_batch_size, num_threads=4, device_id=0)
data = fn.external_source(source=input_data,
cycle=False,
device=device)
processed = fn.coord_flip(data)
pipe.set_outputs(processed)
return pipe
def test_coord_flip_cpu():
pipe = Pipeline(batch_size=batch_size, num_threads=4, device_id=None)
test_data_shape = [200, 2]
def get_data():
out = [
(np.random.randint(0, 255, size=test_data_shape, dtype=np.uint8) /
255).astype(dtype=np.float32) for _ in range(batch_size)
]
return out
data = fn.external_source(source=get_data)
processed = fn.coord_flip(data)
pipe.set_outputs(processed)
pipe.build()
for _ in range(3):
pipe.run()
