本文整理汇总了Python中keras.preprocessing.image.ImageDataGenerator.random_transform方法的典型用法代码示例。如果您正苦于以下问题：Python ImageDataGenerator.random_transform方法的具体用法？Python ImageDataGenerator.random_transform怎么用？Python ImageDataGenerator.random_transform使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类keras.preprocessing.image.ImageDataGenerator的用法示例。
在下文中一共展示了ImageDataGenerator.random_transform方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: ImagePreprocessor
# 需要导入模块: from keras.preprocessing.image import ImageDataGenerator [as 别名]
# 或者: from keras.preprocessing.image.ImageDataGenerator import random_transform [as 别名]
class ImagePreprocessor(object):
"""A Inception v3 image preprocessor. Implements an image augmentation
as well."""
IMAGE_SIZE = (299, 299)
def __init__(self, image_augmentation=None):
self._image_data_generator = ImageDataGenerator(rotation_range=40,
width_shift_range=0.2,
height_shift_range=0.2,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True,
fill_mode='nearest')
if image_augmentation is None:
self._image_augmentation_switch = active_config().image_augmentation
else:
self._image_augmentation_switch = image_augmentation
def preprocess_images(self, img_paths, random_transform=True):
return map(partial(self._preprocess_an_image,
random_transform=random_transform),
img_paths)
def preprocess_batch(self, img_list):
return np.array(img_list)
def _preprocess_an_image(self, img_path, random_transform=True):
img = load_img(img_path, target_size=self.IMAGE_SIZE)
img_array = img_to_array(img)
if self._image_augmentation_switch and random_transform:
img_array = self._image_data_generator.random_transform(img_array)
img_array = inception_v3.preprocess_input(img_array)
return img_array