如何重现 ImageDataGenerator 数据

Question

我是 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 图像并将其转换为该生成器生成的任何内容？我意识到我可以跟踪代码并重现所有内容，但这似乎过于复杂。似乎可能有一些方法或一组方法我没有看到。

Answer 1

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