日々精進

新しく学んだことを書き留めていきます

TensorFlowのTFRecordファイルの中身を確認する方法

作成したTFRecordが想定通りのフォーマットで作られているかを確認したい場合。 tf.data.TFRecordDataset([filepath]) で読み込むと、多分データが必要になるまで読み込まないようになるのでデータを確認出来ず困る(tf.Session()でSessionを開始してsess.runすればいいのかも知れないけど。。) 以下のようにするとJSON形式の文字列として出力させられるので簡単に中身を確認出来る。

    record_iterator = tf.python_io.tf_record_iterator(path="train.tfrecord")
    for string_record in record_iterator:
        example = tf.train.Example()
        example.ParseFromString(string_record)
        print(example)

参考:

www.tensorflow.org