Method __init__

dali/test/python/operator_2/test_python_function.py:53–65

(
        self, batch_size, num_threads, device_id, _seed, image_dir, prefetch_queue_depth=2
    )

class CommonPipeline(Pipeline):
    def __init__(
        self, batch_size, num_threads, device_id, _seed, image_dir, prefetch_queue_depth=2
    ):
        super().__init__(
            batch_size,
            num_threads,
            device_id,
            seed=_seed,
            prefetch_queue_depth=prefetch_queue_depth,
        )
        self.input = ops.readers.File(file_root=image_dir)
        self.decode = ops.decoders.Image(device="cpu", output_type=types.RGB)
        self.resize = ops.PythonFunction(function=resize, output_layouts="HWC")

    def load(self):
        jpegs, labels = self.input()