web-dev-qa-db-ja.com

Tensorflow .tfrecordファイルを検査する方法は？

.tfrecordしかし、私はそれがどのように構成されているのかわかりません。スキーマを調べて、.tfrecordファイルに含まれていますか？

すべてのStackoverflowの回答またはドキュメントは、ファイルの構造を知っていると仮定しているようです。

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

...HOW TO INSPECT serialized_record...
tensorflowtfrecord
30
Bob van Luijt

それを見つけた！

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))
79
Bob van Luijt

https://www.tensorflow.org/programmers_guide/reading_data で指定されているtf.TFRecordReaderデコーダーでTensorFlow tf.parse_single_exampleを使用します

PS、tfrecordには https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/example/example.proto で定義された 'Example'レコードが含まれています

レコードを文字列に抽出すると、解析は次のようになります

a=tf.train.Example()
result = a.ParseFromString(binary_string_with_example_record)

ただし、ファイルから個々のレコードを抽出するための生のサポートがどこにあるのかわかりません。TFRecordReaderで追跡できます。

5
Yaroslav Bulatov