ClassificationImageDataset#

class ClassificationImageDataset[source]#

Bases: ImageDataset

Read images and class labels from images stored in class folders.

I.e., all images for a class “A” are stored in directory A/, all images for a class “B” are stored in directory B/, and so on. And all class directories are located in the same parent directory.

__init__(data_dir: str, class_names: Optional[Iterable[str]], *args, **kwargs)[source]#

Constructor.

Parameters

Methods

__init__(data_dir, class_names, *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__(data_dir: str, class_names: Optional[Iterable[str]], *args, **kwargs)[source]#

Constructor.

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

  • kwargs (Any) –

Return type

Any