我已经使用 Keras 的flow_from_directory方法在非常大的数据集(大约 1200000 个训练图像和 15000 个类)上训练了我的模型。我已经成功地使用fit_generator. 但是,现在我面临predict_generator.
执行时出现以下错误predict_generator:
“找到属于 0 个类别的 0 个图像。”
但是,我已将我的火车图像数据集转换为基于类标签的子目录。但是,对于测试数据集,我没有任何标签。那么我该如何克服这个问题呢?
我已经使用 Keras 的flow_from_directory方法在非常大的数据集(大约 1200000 个训练图像和 15000 个类)上训练了我的模型。我已经成功地使用fit_generator. 但是,现在我面临predict_generator.
执行时出现以下错误predict_generator:
“找到属于 0 个类别的 0 个图像。”
但是,我已将我的火车图像数据集转换为基于类标签的子目录。但是,对于测试数据集,我没有任何标签。那么我该如何克服这个问题呢?
制作文件夹结构。
例如:
Training_Data
|
---Cat
|
-----Dog
因为 keras 期望你的文件夹在 catag