Keras画像前処理

from keras.preprocessing import image
in_path = './dataset'
out_path = 'output'
datagen = image.ImageDataGenerator()
#指定target_size后所有图像都变为相同大小
gen_data = datagen.flow_from_directory(in_path, batch_size=1, shuffle=False,
                                       save_to_dir=out_path+'resize',
                                       save_prefix='gen', target_size=(224, 224))

for i in range(20):
    gen_data.next()