Skip to content

In Memory Pred Dataset

Source

In-memory prediction dataset.

InMemoryPredDataset

Bases: Dataset

Simple prediction dataset returning images along the sample axis.

Parameters:

  • prediction_config (InferenceConfig) –

    Prediction configuration.

  • inputs (NDArray) –

    Input data.

__getitem__(index)

Return the patch corresponding to the provided index.

Parameters:

  • index (int) –

    Index of the patch to return.

Returns:

  • tuple(ndarray, ...) –

    Transformed patch.

__init__(prediction_config, inputs)

Constructor.

Parameters:

  • prediction_config (InferenceConfig) –

    Prediction configuration.

  • inputs (NDArray) –

    Input data.

Raises:

__len__()

Return the length of the dataset.

Returns:

  • int –

    Length of the dataset.