data.load_images_module

A Lightning data-module for loading images recursively from a file-system directory.

Module Contents

Classes

LoadImagesModule

A Lightning data-module for loading images recursively from a file-system directory.
class data.load_images_module.LoadImagesModule(image_directory: str, image_size: int, extension: str = 'jpg', rgb: bool = True, batch_size: int = 16, num_workers: int = 1, ratio_validation: float = 0.3, ratio_test: float = 0.2)[source]

Bases: pytorch_lightning.LightningDataModule

A Lightning data-module for loading images recursively from a file-system directory.

prepare_data(self) → None[source]

Overrides pl.LightningDataModule.

setup(self, stage: Optional[pytorch_lightning.trainer.states.TrainerFn] = None)[source]

Overrides pl.LightningDataModule.

train_dataloader(self)[source]

Overrides pl.LightningDataModule.

val_dataloader(self)[source]

Overrides pl.LightningDataModule.

test_dataloader(self)[source]

Overrides pl.LightningDataModule.

predict_dataloader(self)[source]

Overrides pl.LightningDataModule.