tensorflow - 如何解析单个TFrecord文件

Question

读取 tfrecords:

reader = tf.TFRecordReader()
_, serialized_example = reader.read(filename_queue)
features = tf.parse_single_example(...)

TFRecordReader 从文件队列中读取示例。但是如何从特定文件同步读取单个示例(没有队列)。喜欢

file_buf = tf.read_file(filename)
serialized_example = get_train_example(file_buf)
features = tf.parse_single_example(...)

如何实现get_train_example函数

Answer 1

不确定这是否正是您要找的东西，但您可以在没有队列的情况下这样做:

tf_record = "path/to/my.tfrecord"
e = tf.python_io.tf_record_iterator(tf_record).next()
single_example = tf.parse_single_example(e, features=features)