Image classification

ImageClassificationDataModule

Bases: VisionDataModule[ImageBatchType], ClassificationDataModule[ImageBatchType]

Lightning data modules for image classification.

num_classes instance-attribute

num_classes: int

Number of classes in the dataset.

dims instance-attribute

dims: tuple[C, H, W]

A tuple describing the shape of the data.