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()