Skip to content

semsegdataloader

semsegdataloader

Module for SemSegDataLoader

Classes

SemSegDataLoader 

SemSegDataLoader()

Bases: DataLoader

Implementation of SemSegDataLoader

Source code in niceml/data/dataloaders/dataloader.py 
12
13
def __init__(self):
    self.data_description = None
Functions
load_data 
load_data(data_info)

Takes a SemSegDataInfo object as input, which contains all the information needed to load the image and mask files. The function returns a SemSegData object, which contains both the image and mask data.

Parameters:

  • data_info (SemSegDataInfo) –

    SemSegDataInfo: Contains SemSeg data info

Returns:

Source code in niceml/data/dataloaders/semseg/semsegdataloader.py 
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
def load_data(self, data_info: SemSegDataInfo) -> SemSegData:
    """
    Takes a SemSegDataInfo object as input, which contains all the information needed to load
    the image and mask files. The function returns a SemSegData object, which contains
    both the image and mask data.

    Args:
        data_info: SemSegDataInfo: Contains SemSeg data info

    Returns:
        A SemSegData object

    """
    semseg_dd: SemSegDataDescription = check_instance(
        self.data_description, SemSegDataDescription
    )

    class_lut = semseg_dd.get_class_idx_lut()
    with open_location(data_info.image_location) as (image_fs, image_path):
        image = load_img_uint8(
            image_path,
            file_system=image_fs,
            target_image_size=semseg_dd.get_input_image_size(),
        )
    with open_location(data_info.mask_location) as (mask_fs, mask_path):
        mask_image = load_img_uint8(
            mask_path,
            file_system=mask_fs,
            target_image_size=semseg_dd.get_input_image_size(),
        )

    mask_image = transform_mask_image(mask_image, class_lut)
    return SemSegData(
        file_id=data_info.get_identifier(), image=image, mask_image=mask_image
    )

Functions