加载测试数据以传递到 predict_classes()

数据挖掘 喀拉斯 美国有线电视新闻网
2022-03-06 19:58:27

我正在使用该Sequential模型来训练 CNN。测试数据位于submission_data_path.png 图像中。

是否可以使用ImageDataGenerator'sflow_from_directory加载要传递的测试数据predict_classes()由于flow_from_directory期望图像位于由类命名的文件夹中,有没有办法实现这一点?

或者什么是最首选的方式?谢谢。

这就是我想要做的,

test_data_gen = ImageDataGenerator()
test_it = dataGen.flow_from_directory(submission_data_path, target_size=(64, 64), class_mode='categorical')

predictions = classifier.predict_classes(test_it)
0个回答
没有发现任何回复~