我为Tensorflow构建了一个输入管道来读取我的数据。它被编码为TFRecord文件。我想使用QueueRunners预取数据并将其作为Keras模型的输入。
这些方面的东西:
image_batch, label_batch = MayTFRecordsPipeline.get_batch()
#Keras
model = Sequential()
model.add(Input(tensor=image_batch))
如何使用QueueRunners? 在内部,管道使用(简化)
def get_batch():
filename_queue = tf.train.string_input_producer(self.tfrecord_file_path_list)
tf_keys, decoded_fs = self.decode_multiple_tfrecord(filename_queue)
tensor_list = self.get_multiple_example(decoded_fs)
imageBatch, labelBatch= self.create_joined_shuffled_data_batch(tensor_list)
return image_batch, label_batch