python - 通过ImageDataGenerator类将tfrecord加载到Keras中

Question

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

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

Answer 1

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

查看这篇精彩的文章，了解如何读取 tfrecord 文件:

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