modosaic.segmentation.generators.base.generator

modosaic.segmentation.generators.base.generator

SegmentationGenerator

SegmentationGenerator()

Bases: ModalityGenerator[list[ndarray]], ABC

Base class for segmentation-mask generators.

Attributes:

Name	Type	Description
device	device	Torch device selected for inference.
dtype	dtype	Torch dtype selected for inference.

Initialize device and dtype defaults.

Source code in modosaic/segmentation/generators/base/generator.py

def __init__(self):
    """Initialize device and dtype defaults."""
    self.device, self.dtype = DeviceService.get_device_type()

generate abstractmethod

generate(record)

Generate instance masks for one image record.

Source code in modosaic/segmentation/generators/base/generator.py

@override
@abstractmethod
def generate(self, record: ImageRecord) -> list[np.ndarray]:
    """Generate instance masks for one image record."""
    raise NotImplementedError()