predict_generator 给出“找到属于 0 个类别的 0 个图像”。

数据挖掘 机器学习 Python 深度学习 喀拉斯 图像分类
2022-03-06 07:00:57

我已经使用 Keras 的flow_from_directory方法在非常大的数据集(大约 1200000 个训练图像和 15000 个类)上训练了我的模型。我已经成功地使用fit_generator. 但是,现在我面临predict_generator.

执行时出现以下错误predict_generator

“找到属于 0 个类别的 0 个图像。”

但是,我已将我的火车图像数据集转换为基于类标签的子目录。但是,对于测试数据集,我没有任何标签。那么我该如何克服这个问题呢?

1个回答

制作文件夹结构。

例如:

Training_Data
       |
        ---Cat
       |
       -----Dog

因为 keras 期望你的文件夹在 catag