fruit_project.models.transforms_factory¶

Functions¶

get_transforms(cfg)

Generates a dictionary of Albumentations transformations for training and testing.

Module Contents¶

fruit_project.models.transforms_factory.get_transforms(cfg: omegaconf.DictConfig)[source]¶

Generates a dictionary of Albumentations transformations for training and testing. :param cfg: Configuration object containing the following attributes: :type cfg: DictConfig

Returns:: A dictionary with keys “train” and “test”
Return type:: dict