FrameAugmentation¶

class standard_e2e.dataset_utils.augmentation.FrameAugmentation(*args, **kwargs)[source]¶

Bases: ABC

Abstract base class for per-frame augmentations.

Subclasses implement _augment to transform a mapping of TransformedFrameData keyed by frame name. augment enforces regime validation and delegates to the subclass.

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

augment(frames, regime)[source]¶

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]