通过 ImageDataGenerator 类将 tfrecord 加载到 Keras

【问题标题】：Loading tfrecord into Keras through ImageDataGenerator class通过 ImageDataGenerator 类将 tfrecord 加载到 Keras
【发布时间】：2019-11-21 11:30:27
【问题描述】：

我对 keras 还很陌生，我在这里尝试迁移学习： https://www.tensorflow.org/tutorials/images/transfer_learning

我的数据集不是二进制文件，我有 tfrecord 文件。我可以在 tensorflow 中读取文件。我不想将图像作为输入提供给网络，因为输入来自预训练模型。如何在 Keras 的 ImageDataGenerator 类中传递图像和标签。

【问题讨论】：

标签： python tensorflow keras tfrecord

【解决方案1】：

对于将来可能遇到此问题的任何人。如果预训练过程都是正确的。您可以使用 tf.data API 读取和准备图像以进行训练，并且可以将 (image, label) 集输入到模型的 (.fit) 方法中。

看看这篇很棒的帖子来熟悉如何读取 tfrecord 文件：

https://medium.com/@moritzkrger/speeding-up-keras-with-tfrecord-datasets-5464f9836c36

【讨论】：