TunAugmentor.pipeline module

class TunAugmentor.pipeline.Pipeline(transformations: List[TunAugmentor.transformations.basic_transform.BasicTransform], **kwargs)

Bases: object

Image Augmentation Pipeline. Applies a list of transformations on a list of images. Either applies them to a list of original images or composes the transformations to get 2 to the power of n images.

arrange(images: List[numpy.ndarray):

Composes the transformations to a list of images.

apply(images: List[numpy.ndarray):

Applies the transformations to a list of images