VisionDataset

class VisionDataset(images: Sequence[str | Path | Tensor], transforms: Sequence | None = None, root: Path | None = None)[source]

Bases: Dataset

A dataset of images with data augmentation.

Note

requires torchvision to be installed.

Initialize the dataset.

Parameters:

  • images (Sequence[str | Path | Tensor]) – the images, either as (relative) path, or preprocessed tensors.

  • transforms (Sequence | None) – a sequence of transformations to apply to the images, cf. torchvision.transforms

  • root (Path | None) – the root directory for images