TensorFlow 2 version
|
Set of tools for real-time data augmentation on image data.
Classes
class DirectoryIterator: Iterator capable of reading images from a directory on disk.
class ImageDataGenerator: Generate batches of tensor image data with real-time data augmentation.
class Iterator: Base class for image data iterators.
class NumpyArrayIterator: Iterator yielding data from a Numpy array.
Functions
apply_affine_transform(...): Applies an affine transformation specified by the parameters given.
apply_brightness_shift(...): Performs a brightness shift.
apply_channel_shift(...): Performs a channel shift.
array_to_img(...): Converts a 3D Numpy array to a PIL Image instance.
img_to_array(...): Converts a PIL Image instance to a Numpy array.
load_img(...): Loads an image into PIL format.
random_brightness(...): Performs a random brightness shift.
random_channel_shift(...): Performs a random channel shift.
random_rotation(...): Performs a random rotation of a Numpy image tensor.
random_shear(...): Performs a random spatial shear of a Numpy image tensor.
random_shift(...): Performs a random spatial shift of a Numpy image tensor.
random_zoom(...): Performs a random spatial zoom of a Numpy image tensor.
save_img(...): Saves an image stored as a Numpy array to a path or file object.