Module: tf.keras.preprocessing.image

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.