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

Dan*_*ata 4 python keras tensorflow

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

confusion_matrix(y_true, y_pred)
Run Code Online (Sandbox Code Playgroud)

并且flow_from_directory不返回真实标签。有没有办法flow_from_directory直接或其他任何方式获取它们?

Dan*_*ler 5

对于任何generator,您都可以像处理任何可迭代的那样产生数据:

for x, y in something.flow_from_directory(...):
    #do stuff
Run Code Online (Sandbox Code Playgroud)

小心决定何时停止。在喀拉拉邦训练的生成器是无限的。