Iterator

class dragon.vm.dali.Iterator(pipeline)[source]

The base iterator wrapped for a pipeline.

__init__

Iterator.__init__(pipeline)[source]

Create an Iterator.

Parameters:

Properties

batch_size

Iterator.batch_size

Return the batch size of pipeline.

Returns:
int The batch size.

handlers

Iterator.handlers

Define functions to process pipeline outputs.

Returns:
Sequence[Tuple[Sequence[int], callable]] The handlers.

Methods

get

Iterator.get()[source]

Return the next batch of data.

Alias for self.__next__(...).

Returns:
Sequence[dragon.Tensor] The output tensors.

__next__

Iterator.__next__()[source]

Return the next batch of data.