Python CustomDummy Examples

Example #1

File: demo.py Project: gartnera/dali-numba-plugin

 def __init__(self, batch_size, num_threads, device_id):
     super(SimplePipeline, self).__init__(batch_size,
                                          num_threads,
                                          device_id,
                                          seed=12)
     self.input = ops.FileReader(file_root=image_dir)
     self.decode = ops.ImageDecoder(device='cpu', output_type=types.RGB)
     self.custom = ops.CustomDummy(fn_ptr=hello_cfunc.address)

Example #2

File: test_plugin_manager.py Project: yaysummeriscoming/DALI

 def __init__(self, batch_size, num_threads, device_id):
     super(CustomPipeline, self).__init__(batch_size, num_threads,
                                          device_id)
     self.inputs = ops.ExternalSource()
     self.custom_dummy = ops.CustomDummy(device="gpu")