Iterable Pred Dataset

Iterable prediction dataset used to load data file by file.

`IterablePredDataset`

Bases: IterableDataset

Simple iterable prediction dataset.

Parameters:

prediction_config (InferenceConfig) –

Inference configuration.
src_files (List[Path]) –

List of data files.
read_source_func (Callable, default: read_tiff ) –

Read source function for custom types, by default read_tiff.
**kwargs (Any, default: {} ) –

Additional keyword arguments, unused.

Attributes:

data_path (Union[str, Path]) –

Path to the data, must be a directory.
axes (str) –

Description of axes in format STCZYX.
mean ((Optional[float], optional)) –

Expected mean of the dataset, by default None.
std ((Optional[float], optional)) –

Expected standard deviation of the dataset, by default None.
patch_transform ((Optional[Callable], optional)) –

Patch transform callable, by default None.

Constructor.

Parameters:

prediction_config (InferenceConfig) –

Inference configuration.
src_files (list of pathlib.Path) –

List of data files.
read_source_func (Callable, default: read_tiff ) –

Read source function for custom types, by default read_tiff.
**kwargs (Any, default: {} ) –

Additional keyword arguments, unused.

Raises:

ValueError –

If mean and std are not provided in the inference configuration.

Iterate over data source and yield single patch.

Yields: