semsegdataiterator

semsegdataiterator

Module for semseg prediction container and data handler

Classes

SemSegDataIterator

SemSegDataIterator(
    supported_extensions=None, prediction_suffix="_pred"
)

Bases: TensordataIterator

Data Iterator for SemSeg

Source code in niceml/mlcomponents/resultanalyzers/tensors/semsegdataiterator.py

def __init__(
    self,
    supported_extensions: Optional[List[str]] = None,
    prediction_suffix: str = "_pred",
):
    if supported_extensions is None:
        self.supported_extensions = [".png", ".jpg", ".jpeg", ".tiff", ".bmp"]
    self.data: Optional[dict] = None
    self.prediction_suffix = prediction_suffix
    self.file_system: Optional[AbstractFileSystem] = None

SemSegPredictionContainer dataclass

Contains information about maximum prediction of an image

Functions