SemanticSegmentationImageDataset#

class SemanticSegmentationImageDataset[source]#

Bases: ImageDataset

Reads semantic segmentatioin images and labels from files.

Uses SemanticSegmentationDataReader to read the data.

__init__(img_dir: str, label_dir: str, *args, **kwargs)[source]#

Constructor.

Parameters

Methods

__init__(img_dir, label_dir, *args, **kwargs)

Constructor.

__contains__(key: str) bool#
Parameters

key (str) –

Return type

bool

__getitem__(key) Tuple[torch.Tensor, torch.Tensor]#
Return type

Tuple[torch.Tensor, torch.Tensor]

__init__(img_dir: str, label_dir: str, *args, **kwargs)[source]#

Constructor.

Parameters
static __new__(cls, *args: Any, **kwargs: Any) Any#
Parameters
  • args (Any) –

  • kwargs (Any) –

Return type

Any