Keras的Tensorflow输入管道(TFRecords)

时间:2017-05-24 17:49:11

标签: python input tensorflow keras pipeline

我为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

0 个答案:

没有答案