繁体   English   中英

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

[英]Keras: Obtain array of images and labels from flow_from_directory

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

confusion_matrix(y_true, y_pred)

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

对于任何generator ,您可以像任何可迭代的那样generator数据:

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

小心决定何时停止。 用于训练keras的发生器是无限的。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM