你可以使用 tf.python_io.tf_record_iterator
来迭代读取 tf.records
中的每一条记录。
下面是一个示例代码:
import tensorflowas tf
# 创建一个读取器
reader = tf.TFRecordReader()
# 打开要读取的 tfrecord 文件
filename_queue = tf.train.string_input_producer(['path/to/your.tfrecord'])
# 解码
_, serialized_example = reader.read(filename_queue)
features = tf.parse_single_example(serialized_example, features={
'feature1': tf.FixedLenFeature([], tf.int64),
'feature2': tf.FixedLenFeature([], tf.string),
# ...
})
# 在使用的时候,你可以通过调用 sess.run(features) 来查看每一条记录的内容
这样你就可以解码 tf.records
文件中的每一条记录了。