pyalfe.inference

pyalfe.inference#

class pyalfe.inference.InferenceModel[source]#

Bases: ABC

The parent class for all inference models.

abstract predict_cases(input_image_tuple_list: list[tuple], output_list: list) → None[source]#
Parameters:

  • input_image_tuple_list

  • output_list

class pyalfe.inference.NNUnet(model_dir: str, fold: int, n_threads_preprocessing: int = 6, n_threads_save: int = 1)[source]#

Bases: InferenceModel

predict_cases(input_image_tuple_list, output_list)[source]#
Parameters:

  • input_image_tuple_list

  • output_list

class pyalfe.inference.NNUnetV2(model_dir: str, folds: tuple[str] | None = None, n_threads_preprocessing: int = 1, n_threads_save: int = 1)[source]#

Bases: InferenceModel

predict_cases(input_image_tuple_list, output_list)[source]#
Parameters:

  • input_image_tuple_list

  • output_list

class pyalfe.inference.SynthSeg[source]#

Bases: InferenceModel

predict_cases(input_image_tuple_list, output_list)[source]#
Parameters:

  • input_image_tuple_list

  • output_list