src.FRAME_FM.datasets.InputOnly_Dataset

Lightweight Dataset wrapper that takes inputs only

Classes

TransformedInputDataset	This class applies to input only style datasets that are useful for visual
TransformedInputCoordsDataset	This class applies to input only in the dataset that also passes grid tile coordinates.

Module Contents

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

Bases: torch.utils.data.Dataset

This class applies to input only style datasets that are useful for visual autoencoders. The method currently uses a scaling coefficient to scale the input, this will change in the future when the decision about transform settings are finalized.

base

transform = None

__len__() → int

__getitem__(idx: int)

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

Bases: torch.utils.data.Dataset

This class applies to input only in the dataset that also passes grid tile coordinates.

base

transform = None

__len__() → int

__getitem__(idx: int)