小编J.K*_*Kim的帖子

在使用 tf.keras.preprocessing.image_dataset_from_directory() 时如何在预测期间获取文件名?

Keras最近引入了tf.keras.preprocessing.image_dataset_from_directory函数,比之前tensorflow 2.x 中的ImageDataGenerator.flow_from_directory方法效率更高。

我正在练习 catvsdogs 问题并使用此函数为我的模型构建数据管道。训练模型后,我使用 preds = model.predict(test_ds) 来获取我的测试数据集的预测。我应该如何将pred与图片名称匹配?(之前有 generator.filenames,但在新方法中不再存在。)谢谢!

dataset prediction keras tensorflow

8
推荐指数
3
解决办法
2047
查看次数

标签 统计

dataset ×1

keras ×1

prediction ×1

tensorflow ×1