neural-network - 使用 keras 上的目录测试模型

Question

所以我使用 keras 库训练了一个 CNN，特别是使用 .flow_from_directory()方法，我的图像存储在 training_set , validation_set和 test_set文件夹，但我找不到测试 test_set 的方法文件夹。

我必须创建一个 .csv 文件吗？

我试过使用 model.evaluate() ，但这似乎不适用于存储在分类文件夹中的图像。在此先感谢您的帮助!

Answer 1

您可以使用 evaluate_generator功能。您可以使用相同的 flow_from_directory提供生成器作为此函数的第一个参数。

test_datagen = ImageDataGenerator(...)
test_generator = test_datagen.flow_from_directory(...)
model.evaluate_generator(test_generator, ...)