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

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

最佳答案

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

看看这篇很棒的文章,以熟悉如何阅读tfrecord文件:

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

关于python - 通过ImageDataGenerator类将tfrecord加载到Keras中,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/56994396/

10-12 15:56