python - Keras:从 flow_from_directory 获取图像和标签数组

Question

我在训练模型后尝试使用 scikit 获取混淆矩阵，但由于我使用了 flow_from_directory，所以我无法访问数据和标签，或者我不知道方法这样做。由于 scikit 混淆矩阵方法的使用方式如下:

confusion_matrix(y_true, y_pred)

和flow_from_directory 不返回真实标签。有没有办法直接从 flow_from_directory 或任何其他方法获取它们？

Answer 1

对于任何生成器，您都可以像任何可迭代对象一样生成数据:

for x, y in something.flow_from_directory(...):
    #do stuff

谨慎决定何时停止。用于在 keras 中训练的生成器是无限的。