semsegdatainfolisting

semsegdatainfolisting

Module for SemSegDataInfoListing

Classes

SemSegDataInfoListing

SemSegDataInfoListing(
    location,
    sub_dir,
    mask_suffix="_mask",
    img_suffixes=None,
)

Bases: DataInfoListing

Lists all image files which have a corresponding mask suffix

Source code in niceml/data/datainfolistings/semsegdatainfolisting.py

def __init__(
    self,
    location: Union[dict, LocationConfig],
    sub_dir: str,
    mask_suffix: str = "_mask",
    img_suffixes: Optional[List[str]] = None,
):
    self.mask_suffix = mask_suffix
    self.img_suffixes = img_suffixes or [".png", ".jpg", ".jpeg"]
    self.sub_dir = sub_dir
    self.location = location

Functions