dcnum.segm.segm_torch.segm_torch_mpo

Classes

SegmentTorchMPO

Base segmenter class

Module Contents

class dcnum.segm.segm_torch.segm_torch_mpo.SegmentTorchMPO(*, kwargs_mask: dict | None = None, debug: bool = False, **kwargs)[source]

Bases: dcnum.segm.segm_torch.segm_torch_base.TorchSegmenterBase, dcnum.segm.segmenter_mpo.MPOSegmenter

Base segmenter class

This is the base segmenter class for the multiprocessing operation segmenter segmenter_mpo.MPOSegmenter (multiple subprocesses are spawned and each of them works on a queue of images) and the single-threaded operation segmenter segmenter_sto.STOSegmenter (e.g. for batch segmentation on a GPU).

Parameters:

  • kwargs_mask (dict) – Keyword arguments for mask post-processing (see process_labels)

  • debug (bool) – Enable debugging mode (e.g. CPU segmenter runs in one thread)

  • kwargs – Additional, optional keyword arguments for segment_batch.

static segment_algorithm(image, *, model_file: str | None = None)[source]
Parameters:

  • image (2d ndarray) – event image

  • model_file (str) – path to or name of a dcnum model file (.dcnm); if only a name is provided, then the “torch_model_files” directory paths are searched for the file name

Returns:

mask – mask or labeling image for the give index

Return type:

2d boolean or integer ndarray