FRAME_FM.dataloaders package

Submodules

FRAME_FM.dataloaders.combined_dataloader module

class FRAME_FM.dataloaders.combined_dataloader.CombinedDataModule(datamodules: list[BaseDataModule], batch_size: int = 32, num_workers: int = 4, pin_memory: bool = True, persistent_workers: bool = False, split_strategy: str = 'fraction', train_split: float = 0.85, val_split: float = 0.15, test_split: float = 0.0, train_transforms: Callable | None = None, val_transforms: Callable | None = None, test_transforms: Callable | None = None)[source]

Bases: BaseDataModule

A DataModule for combining datasets from multiple DataModules. # TODO: Implement post-combination transforms

FRAME_FM.dataloaders.demo_eurosat module

class FRAME_FM.dataloaders.demo_eurosat.EuroSATDataModule(data_root: str = 'data', batch_size: int = 32, num_workers: int = 4, pin_memory: bool = True, persistent_workers: bool = False, **kwargs: Any)[source]

Bases: BaseDataModule

FRAME-FM DataModule for EuroSAT.

  • Uses BaseDataModule for split logic (split_strategy, indices/fractions).

  • Uses Hydra-provided transforms (train_transforms, val_transforms, test_transforms).

  • Wraps split datasets in TransformedDataset so each split can have its own transform.

prepare_data() → None[source]

Download EuroSAT once.

Module contents