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
img_dir (str) – Directory containing images.
label_dir (str) – Directory containing segmentation masks.
*args – See
ImageDataset.__init__()
.**kwargs – See
ImageDataset.__init__()
.
Methods
__init__
(img_dir, label_dir, *args, **kwargs)Constructor.
- __getitem__(key) Tuple[torch.Tensor, torch.Tensor] #
- Return type
- __init__(img_dir: str, label_dir: str, *args, **kwargs)[source]#
Constructor.
- Parameters
img_dir (str) – Directory containing images.
label_dir (str) – Directory containing segmentation masks.
*args – See
ImageDataset.__init__()
.**kwargs – See
ImageDataset.__init__()
.