通过ImageDataGenerator类将tfrecord加载到Keras中

Question

我对keras相当陌生，我正在尝试在这里转移学习： https://www.tensorflow.org/tutorials/images/transfer_learning

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

Answer 1

对于任何将来可能有这个问题的人。如果预训练过程都是正确的。你可以使用tf.数据API用于读取和准备用于训练的图像和（图像，标签）集，可以将其馈送到模型的（.fit）方法。在

请看这篇伟大的文章，以熟悉如何读取tfrecord文件：

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