如何重现ImageDataGenerator数据

时间:2018-03-12 13:24:28

标签: python numpy machine-learning keras

我对Keras很新,并使用ImageDataGenerator来训练目录中的图像。我现在想要在另一个目录中预测未知图像,但我不确定如何重现以下代码正在进行的任何转换。

generator = ImageDataGenerator(rescale=1. / 255).flow_from_directory(
    train_data_dir,
    target_size=(img_width, img_height),
    batch_size=16,
    class_mode='binary')

有没有办法采取任意numpy数组或pil图像并将其转换为此生成器生成的任何内容?我意识到我可以跟踪代码并重现所有内容,但这似乎过于复杂。似乎可能有一些我没有看到的方法或方法。

1 个答案:

答案 0 :(得分:0)

从上下文中,您可能希望将代码中的train_data_dir替换为要从中获取图像的other_dir