Module: tf.keras.preprocessing.image

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.

smart_resize(...): Resize images to a target size without aspect ratio distortion.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/keras/preprocessing/image