fruit_project.models.transforms_factory

Functions

get_transforms(→ Dict[str, albumentations.Compose])	Generates a dictionary of Albumentations transformations for training and testing.
get_bbox_params(cfg)

Module Contents

fruit_project.models.transforms_factory.get_transforms(cfg: omegaconf.DictConfig, id2label: Dict[int, str]) → Dict[str, albumentations.Compose]

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

fruit_project.models.transforms_factory.get_bbox_params(cfg)