Microsplit Data Config

Pretty string reprensenting the configuration.

Returns:

• str –

  Pretty string.

Validate axes.

Axes must: - be a combination of 'STCZYX' - not contain duplicates - contain at least 2 contiguous axes: X and Y - contain at most 4 axes - not contain both S and T axes

Parameters:

• axes (str) –

  Axes to validate.

• info (ValidationInfo) –

  Validation information.

Returns:

• str –

  Validated axes.

Raises:

• ValueError –

  If axes are not valid.

Validate that batch_size is not set in the dataloader parameters.

batch_size must be set through batch_size field, not through the dataloader parameters.

Parameters:

• dataloader_params (dict of {str: Any}) –

  The dataloader parameters.

Returns:

• dict of {str: Any} –

  The validated dataloader parameters.

Raises:

• ValueError –

  If batch_size is present in the dataloader parameters.

Convert mode while preserving MicroSplit-specific fields.

Parameters:

• new_mode (Literal['validating', 'predicting']) –

  The new dataset mode, one of validating or predicting.

• new_patch_size (Sequence[int] or None, default: None ) –

  New patch size. If None for predicting, uses whole image prediction.

• overlap_size (Sequence[int] or None, default: None ) –

  New overlap size. Required when switching to tiled prediction with new_patch_size.

• new_batch_size (int or None, default: None ) –

  New batch size. If None, keeps the current batch size.

• new_data_type ((array, tiff, zarr, czi, custom), default: "array" ) –

  New data type. If None, keeps the current data type.

• new_axes (str or None, default: None ) –

  New axes. If None, keeps the current axes.

• new_channels ((Sequence[int], all or None), default: None ) –

  New channel selection. If None, keeps the current channel selection. If "all", selects all channels.

• new_in_memory (bool or None, default: None ) –

  New in-memory loading setting. If None, keeps the current setting.

• new_dataloader_params (dict[str, Any] or None, default: None ) –

  New dataloader parameters for the converted mode.

Returns:

• MicroSplitDataConfig –

  Converted configuration with relevant MicroSplit-specific fields preserved.

Check if the data is 3D based on the axes.

Either "Z" is in the axes and patching patch_size has 3 dimensions, or for CZI data, "Z" is in the axes or "T" is in the axes and patching patch_size has 3 dimensions.

This method is used during Configuration validation to cross checks dimensions with the algorithm configuration.

Returns:

• bool –

  True if the data is 3D, False otherwise.

Propagate the main seed to all augmentations that support seeds.

This ensures that all augmentations use the same seed for reproducibility, unless they already have a seed explicitly set.

Returns:

• Self –

  Data model with propagated seeds.

Propagate the main seed to the patching strategy if it supports seeds.

This ensures that the patching strategy uses the same seed for reproducibility, unless it already has a seed explicitly set.

Returns:

• Self –

  Data model with propagated seed.

Raise error for features not supported by MicroSplit.

Returns:

• Self –

  Validated config.

Set 3D parameters.

Parameters:

• axes (str) –

  Axes.

• patch_size (list of int) –

  Patch size.

Set default max patch filter coverage based on data dimensionality.

Returns:

• Self –

  Data model with default max patch filter coverage updated.

Set default pin_memory for dataloader parameters if not provided.

• If 'pin_memory' is not set, it defaults to True if CUDA is available.

Parameters:

• dataloader_params (dict of {str: Any}) –

  The dataloader parameters.

Returns:

• dict of {str: Any} –

  The dataloader parameters with pin_memory default applied.

Set num_workers and persistent_workers defaults in all dataloaders.

For each of train_dataloader_params, val_dataloader_params, and pred_dataloader_params: sets num_workers from the num_workers field if not already present, and sets persistent_workers=True when num_workers > 0 and not already specified.

Returns:

• Self –

  Validated data model with worker defaults applied to all dataloaders.

Validate that "shuffle" is included in the training dataloader params.

A warning will be raised if shuffle=False.

Parameters:

• train_dataloader_params (dict of {str: Any}) –

  The training dataloader parameters.

Returns:

• dict of {str: Any} –

  The validated training dataloader parameters.

Raises:

• ValueError –

  If "shuffle" is not included in the training dataloader params.

Validate channels.

Channels must be a sequence of non-negative integers without duplicates. If channels are not None, then C must be present in the axes.

Parameters:

• channels (Sequence of int or None) –

  Channels to validate.

• info (ValidationInfo) –

  Validation information.

Returns:

• Sequence of int or None –

  Validated channels.

Raises:

• ValueError –

  If channels are not valid.

Validate 2D/3D dimensions between axes and patch size.

Returns:

• Self –

  Validated data model.

Raises:

• ValueError –

  If the patch size dimension is not compatible with the axes.

Validate that the filters are only used during training.

Parameters:

• filter_obj (PatchFilterConfig | MaskFilterConfig | None) –

  Filter to validate.

• info (ValidationInfo) –

  Validation information.

Returns:

• PatchFilterConfig | MaskFilterConfig | None –

  Validated filter.

Raises:

• ValueError –

  If a filter is used in a mode other than training.

Validate that in_memory is compatible with data_type.

in_memory can only be True for 'array', 'tiff' and 'custom' data types.

Parameters:

• in_memory (bool) –

  Whether to load data into memory.

• info (Any) –

  Additional information about the field being validated.

Returns:

• bool –

  Validated in_memory value.

Raises:

• ValueError –

  If in_memory is True for unsupported data types.

Validate certain parameters are not set for prediction.

Returns:

• Self –

  Validated config.

Validate that the patching strategy is compatible with the dataset mode.

• If mode is training, patching strategy must be random or stratified.
• If mode is validating, patching must be fixed_random.
• If mode is predicting, patching strategy must be tiled or whole.

Parameters:

• patching (PatchingStrategies) –

  Patching strategy to validate.

• info (ValidationInfo) –

  Validation information.

Returns:

• PatchingStrategies –

  Validated patching strategy.

Raises:

• ValueError –

  If the patching strategy is not compatible with the dataset mode.

Warn if num_workers conflicts with a per-dataloader value.

This validator runs before set_default_workers_in_dataloaders, so the dataloader dicts only contain user-supplied values at this point. Only fires when num_workers was explicitly set on the model.

Returns:

• Self –

  Unchanged data model.