Keras引入了TF。keras.preprocessing.image_dataset_from_directory函数,它比以前的ImageDataGenerator更高效。tensorflow 2中的flow_from_directory方法。x.
我正在练习catsvsdogs问题,并使用此函数为我的模型构建数据管道。在训练模型之后,我使用preds = model。predict(test_ds)获取我的测试数据集的预测。我应该如何匹配图片名称的preds?(有发电机。文件名之前,但在新方法中不再存在。)谢谢!
4条答案
按热度按时间zu0ti5jz1#
扩展@丹尼尔Woolcott和@Almog大卫的答案,文件路径由Tensorflow v2中的
image_dataset_from_directory()
函数返回。4.已经。不需要改变函数的源代码。更准确地说,您可以轻松地使用
file_paths
属性检索路径。试试这个:
打印输出:
a7qyws3x2#
我也遇到过类似的问题。解决方案是获取底层的tf。keras.preprocessing.image_dataset_from_directory函数并将'image_paths'变量添加到return语句。这不会引起计算开销,因为已经检索了文件名。
主函数代码取自GitHub: www.example.com
参见下文:
其工作方式为:
train_paths返回文件字符串列表。
bhmjp9jg3#
Tensorflow 24数据集有一个名为的字段:
file_paths
因此可以使用它来获取文件路径。如果您在数据集创建中使用
shuffle=True
,请注意,您必须在数据集创建代码中禁用此行(方法:image_dataset_from_directory
):irlmq6kh4#
基于此https://datascience.stackexchange.com/a/93308/149241
更多细节可以在这里以及https://kylewbanks.com/blog/loading-unlabeled-images-with-imagedatagenerator-flowfromdirectory-keras