作成した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)
参考: