from keras.preprocessing.image import load_img, img_to_array
from keras.preprocessing.image import ImageDataGenerator
import matplotlib.pyplot as plt
import numpy as np
import os
import glob
 
input_dir = "in_dir"
files = glob.glob(input_dir + '/*.png')
 
output_dir = "out_dir"
if os.path.isdir(output_dir) == False:
    os.mkdir(output_dir)
 
for i, file in enumerate(files):
 
    img = load_img(file)
    x = img_to_array(img)
    x = np.expand_dims(x, axis=0)
 
    datagen = ImageDataGenerator(
        rotation_range=0,
        width_shift_range=0,
        height_shift_range=0)
 
    g = datagen.flow(x, batch_size=1, save_to_dir=output_dir, save_prefix='img', save_format='jpg')
    for i in range(1):
        batch = g.next()