Skip to content

Iterable Pred Dataset

Source

Iterable prediction dataset used to load data file by file.

IterablePredDataset

Bases: IterableDataset

Simple iterable prediction dataset.

Parameters:

Name Type Description Default
prediction_config InferenceConfig

Inference configuration.

required
src_files List[Path]

List of data files.

required
read_source_func Callable

Read source function for custom types, by default read_tiff.

read_tiff
**kwargs Any

Additional keyword arguments, unused.

{}

Attributes:

Name Type Description
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.