Skip to content
heliaEDGE
Reference
HELIA

augmentation_pipeline

Serializable composition of preprocessing and training-only augmentation.

Machine-readable model

class

Apply layers in order to a tensor or structured sample.

helia_edge/layers/preprocessing/augmentation_pipeline.py:8

AugmentationPipeline(layers: list[keras.Layer], name=None, force_training=False, **kwargs={})

Apply layers in order to a tensor or structured sample.

Each child receives the preceding child’s output and the training flag. Layers must accept training and, when supplied, transformations. Calling with transformations requires one entry per layer; use None for a child that should sample its own parameters.

Parameters of AugmentationPipeline
NameTypeDefaultDescription
layerslist[keras.Layer]RequiredOrdered layers to apply. The layer objects are retained.
namestr | NoneNoneOptional Keras layer name.
force_trainingboolFalsePass training=True to every child, including at inference.
**kwargsAny{}Additional Keras Layer arguments.
Errors raised by AugmentationPipeline
TypeDescription
ValueErrorThe supplied transformations count differs from the layer count.