我对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/