我有一个.tfrecord,但我不知道它的结构。如何检查架构以了解.tfrecord文件包含什么?

所有Stackoverflow答案或文档似乎都假设我知道文件的结构。

reader = tf.TFRecordReader()
file = tf.train.string_input_producer("record.tfrecord")
_, serialized_record = reader.read(file)

...HOW TO INSPECT serialized_record...

最佳答案

找到了!

import tensorflow as tf

for example in tf.python_io.tf_record_iterator("data/foobar.tfrecord"):
    print(tf.train.Example.FromString(example))

您还可以添加:
from google.protobuf.json_format import MessageToJson
...
jsonMessage = MessageToJson(tf.train.Example.FromString(example))

关于tensorflow - 如何检查Tensorflow .tfrecord文件?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/42394585/

10-11 18:05