如何解析单个 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)