MultipleFramesImageAugmentation

class standard_e2e.dataset_utils.augmentation.MultipleFramesImageAugmentation(regime, train_transform=None, val_transform=None, *args, **kwargs)

Bases: FrameAugmentation

Apply an Albumentations pipeline to each camera modality in given frames.

Parameters:

• regime (str)

• train_transform (Compose | None)

• val_transform (Compose | None)

ALLOWED_REGIMES = ['train', 'val', 'test']

augment(frames, regime)

Apply augmentation to the given frame data.

Parameters:

• frames (dict[str, TransformedFrameData]) – The frame data to augment.

• regime (str) – The regime for which the augmentation is applied.

Returns:

The augmented frame data.

Return type:

dict[str, TransformedFrameData]