mmv_im2im.configs package

Submodules

mmv_im2im.configs.config_base module

class mmv_im2im.configs.config_base.ArgumentParser(config_class: ~typing.Type[~mmv_im2im.configs.config_base.T], config: str | None = None, formatter_class: ~typing.Type[~argparse.HelpFormatter] = <class 'pyrallis.help_formatter.SimpleHelpFormatter'>, *args, **kwargs)

Bases: Generic[T], ArgumentParser

Creates an ArgumentParser instance.

parse_args(args=None, namespace=None) → T

parse_known_args(args: Sequence[str] | None = None, namespace: Namespace | None = None, attempt_to_reorder: bool = False)

print_help(file=None)

set_dataclass(dataclass: ~typing.Type[~pyrallis.utils.Dataclass] | ~pyrallis.utils.Dataclass, prefix: str = '', default: ~pyrallis.utils.Dataclass | ~typing.Dict | None = None, dataclass_wrapper_class: ~typing.Type[~pyrallis.wrappers.dataclass_wrapper.DataclassWrapper] = <class 'pyrallis.wrappers.dataclass_wrapper.DataclassWrapper'>)

Adds command-line arguments for the fields of dataclass.

class mmv_im2im.configs.config_base.DataConfig(category: str | None = None, data_path: ~pathlib.Path | str | ~typing.Dict | None = None, cache_path: str | ~pathlib.Path | None = None, inference_output: ~mmv_im2im.configs.config_base.InferOutConfig = <factory>, inference_input: ~mmv_im2im.configs.config_base.InferInConfig = <factory>, dataloader: ~mmv_im2im.configs.config_base.DataloaderConfig = <factory>, preprocess: ~typing.List[~typing.Dict] | None = None, postprocess: ~typing.List[~typing.Dict] | None = None, augmentation: ~typing.List[~typing.Dict] | None = None, patch_size: ~typing.List = <factory>, extra: ~typing.Dict = <factory>)

Bases: object

Top-level config for data: major components

augmentation: List[Dict] = None

cache_path: Path | str = None

category: str = None

data_path: Path | str | Dict = None

dataloader: DataloaderConfig

extra: Dict

inference_input: InferInConfig

inference_output: InferOutConfig

patch_size: List

postprocess: List[Dict] = None

preprocess: List[Dict] = None

class mmv_im2im.configs.config_base.DataloaderConfig(train_val_ratio: float = 0.1, train: ~mmv_im2im.configs.config_base.DataloaderModuleConfig = <factory>, val: ~mmv_im2im.configs.config_base.DataloaderModuleConfig = <factory>)

Bases: object

Mid-level config for data: about dataloader module

train: DataloaderModuleConfig

train_val_ratio: float = 0.1

val: DataloaderModuleConfig

class mmv_im2im.configs.config_base.DataloaderModuleConfig(dataloader_type: ~typing.Dict = <factory>, dataset_params: ~typing.Dict = <factory>, dataloader_params: ~typing.Dict = <factory>, partial_loader: float = 1.0)

Bases: object

low-level config for data: specifically about the dataloader

dataloader_params: Dict

dataloader_type: Dict

dataset_params: Dict

partial_loader: float = 1.0

class mmv_im2im.configs.config_base.InferInConfig(dir: Path | str | Dict | None = None, data_type: str = 'tiff', reader_params: Dict | None = None)

Bases: object

config for the input to inference

data_type: str = 'tiff'

dir: Path | str | Dict = None

reader_params: Dict = None

class mmv_im2im.configs.config_base.InferOutConfig(path: Path | str | Dict | None = None, suffix: str = '_pred')

Bases: object

config for the output of inference

path: Path | str | Dict = None

suffix: str = '_pred'

class mmv_im2im.configs.config_base.ModelConfig(framework: str | None = None, net: ~typing.Dict = <factory>, criterion: ~typing.Dict = <factory>, optimizer: ~typing.Dict = <factory>, scheduler: ~typing.Dict = <factory>, checkpoint: str | ~pathlib.Path | None = None, model_extra: ~typing.Dict = <factory>)

Bases: object

Config for model

checkpoint: Path | str = None

criterion: Dict

framework: str = None

model_extra: Dict

net: Dict

optimizer: Dict

scheduler: Dict

class mmv_im2im.configs.config_base.ProgramConfig(mode: str | None = None, data: ~mmv_im2im.configs.config_base.DataConfig = <factory>, model: ~mmv_im2im.configs.config_base.ModelConfig = <factory>, trainer: ~mmv_im2im.configs.config_base.TrainerConfig = <factory>)

Bases: object

the main configuration for the whole program

data: DataConfig

mode: str = None

model: ModelConfig

trainer: TrainerConfig

class mmv_im2im.configs.config_base.TrainerConfig(verbose: bool = False, params: ~typing.Dict | None = None, callbacks: ~typing.List[~typing.Dict] = <factory>, gpus: int | ~typing.List[int] = <factory>)

Bases: object

Config for how to run the training

callbacks: List[Dict]

gpus: int | List[int]

params: Dict = None

verbose: bool = False

mmv_im2im.configs.config_base.configuration_validation(cfg)

mmv_im2im.configs.config_base.parse_adaptor(config_class: Type[T], config: str | Path | None = None, args: Sequence[str] | None = None) → T

Module contents