generator Namespace Reference

Classes
class	CustomDataGenerator

Functions
	get_augmenter (input_size=(224, 224))

Function Documentation

get_augmenter()

get_augmenter

(

input_size = (224, 224)

)

Create an image augmenter using the imgaug library.

Args:
    input_size (tuple): Desired output size of the images (height, width).

Returns:
    iaa.Sequential: Augmentation pipeline.