tensorflow - 将迭代器(来自 tf.data.Dataset)中的元素提供给 TensorFlow 模型的有效方法是什么?

标签 tensorflow tensorflow-datasets

我正在使用 TensrFlow 的新 API 通过 tf.data.Dataset 和迭代器导入数据。它工作正常,但我不确定我所做的是否有效。

我现在正在做的是评估迭代器的 get_next() 方法,它为我提供了一堆元素,例如实际图像、标签、文件名等。然后我将使用 feed_dict 将图像导入我的模型。

我知道 feed_dict 非常慢,所以我是否失去了 Dataset 和 Iterators 的好处,并通过评估在 TFRecord 中序列化了数据集条目并通过 feed_dict 将它们馈送到图中?我没有在 TF 的文档中找到任何示例,这些示例说明了人们期望如何使用 Iterator 的 get_next() 将元素馈送到模型中。解压 get_next() 并直接在我的图表中使用结果是否更好?

最佳答案

是的,建议在您的图表中直接使用 get_next() 返回的张量。也许你错过了 Programmer's Guidetf.data API 上。

您还可以找到 performance guide有指导意义。

希望对您有所帮助。

关于tensorflow - 将迭代器(来自 tf.data.Dataset)中的元素提供给 TensorFlow 模型的有效方法是什么?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48427269/

相关文章:

tensorflow - 相同的神经网络权重不应该产生相同的结果吗?

tensorflow - 如何将多输入图像应用于Unet进行分割?

python - 优化性能的 tf.data 管道设计

python - 为什么 tf.keras.layers.Conv2D 在每次运行时给出不同的结果

python - 如何使用 Tensorflow 2/Keras 保存和继续训练具有多个模型部分的 GAN

python - 仅对图像应用预处理,而不对蒙版应用预处理

python - 如何将 tensorflow-hub 模块与 tensorflow-dataset api 一起使用

python - 如何在完整验证示例上评估 Tensorflow 模型

tensorflow - 如何使用 tf.MonitoredTrainingSession 在训练和验证数据集之间切换?

python - ValueError : Dimensions must be equal,,但对于 'loss/output_1_loss/mul' 来说是 3 和 3072 (op : 'Mul' ) with input shapes: [? ,3], [?,3072]