Загрузка tfrecord в Keras через класс ImageDataGenerator

Question

Я довольно новичок в keras и пытаюсь перенести обучение здесь: https://www.tensorflow.org/tutorials/images/transfer_learning

Мой набор данных, однако, не является двоичным, и у меня есть файл tfrecord.Я могу прочитать файл в tenorflow.Я не хочу подавать изображения в качестве входных данных в сеть, поскольку входные данные поступают от предварительно обученной модели.Как можно передать изображения и метки в классе ImageDataGenerator в Keras.

Answer 1

Для тех, кто может иметь эту проблему в будущем. Если перед тренировкой все в порядке. Вы можете использовать API tf.data для чтения и подготовки изображений для обучения, а набор (image, label) можно подавать в метод (.fit) вашей модели.

Посмотрите на этот великий пост, чтобы узнать, как читать файл tfrecord:

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