SemanticSegmentationDataReader

class SemanticSegmentationDataReader

Bases: Dataset

Reads semantic segmentatioin images and labels from files.

__init__(img_dir: str, label_dir: str)

Constructor.

Parameters

• img_dir (str) – Directory containing images.

• label_dir (str) – Directory containing segmentation masks.

Methods

__init__(img_dir, label_dir)	Constructor.
validate_paths()

__contains__(key: str) → bool

Parameters

key (str) –

Return type

bool

__getitem__(ind: int) → Tuple[ndarray, ndarray]

Parameters

ind (int) –

Return type

Tuple[ndarray, ndarray]

__init__(img_dir: str, label_dir: str)

Constructor.

Parameters

• img_dir (str) – Directory containing images.

• label_dir (str) – Directory containing segmentation masks.

static __new__(cls, *args: Any, **kwargs: Any) → Any

Parameters

• args (Any) –

• kwargs (Any) –

Return type

Any

validate_paths() → None

Return type

None