如何从 ImageDataGenerator 获取元组?

时间:2021-01-11 15:41:45

标签: python keras generator

我一直在许多示例中使用 ImageDataGenerator 和 flow_from_directory:

val_gen = ImageDataGenerator()
val_imageflow = val_gen.flow_from_directory(validation_path, ...) 

然后在 fit 函数中使用它来训练和验证数据:

hist = model.fit(..., validation_data=val_imageflow,...)

但现在我需要在形状 (x_data, y_data) 中使用validation_data,例如案例

hist = model.fit(..., validation_data=(x_data, y_data),...)

原因是我想在每个时代结束时在我微调的 CNN 中使用回调来显示每个类的正确分类。不幸的是,我无法理解如何将生成器 val_imageflow 放入元组。我试图运行以下代码,但它很快消耗了所有 RAM 并且处理被中断(我可能做错了什么)

x_data = []
y_data = []
while True:
    try:
        n = next(val_imageflow)
        x_data.append(n[0])
        y_data.append(n[1])
    except StopIteration:
        break

0 个答案:

没有答案
相关问题