src.FRAME_FM.datasets.ImageLabel_Dataset

Lightweight Dataset wrapper that applies transforms to images only, preserving the (image, label) structure of torchvision datasets.

Classes

TransformedDataset

PyTorch Dataset wrapper that applies transforms to images only,

Module Contents

class src.FRAME_FM.datasets.ImageLabel_Dataset.TransformedDataset(base: torch.utils.data.Dataset, transform: Any | None = None)

Bases: torch.utils.data.Dataset

PyTorch Dataset wrapper that applies transforms to images only,

base

transform = None

__len__() → int

__getitem__(idx: int)