TensorFlowでkeras その4


概要

TensorFlowでkerasやってみた。
ImageDataGeneratorやってみた。

写真

環境

windows 7 sp1 64bit
anaconda3
tensorflow 1.2

サンプルコード


from tensorflow.contrib.keras.python.keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array, load_img

datagen = 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')
img = load_img('./p1m.jpg')
x = img_to_array(img)
x = x.reshape((1, ) + x.shape)
i = 0
for batch in datagen.flow(x, batch_size = 1, save_to_dir = './', save_prefix = 't', save_format = 'jpeg'):
    i += 1
    if i > 20:
        break