tf1.x和tf2.x查看TFRecord数据的方法

最新推荐文章于 2024-04-28 22:50:14 发布

江南蜡笔小新

最新推荐文章于 2024-04-28 22:50:14 发布

阅读量669

点赞数

分类专栏：杂记 Note 文章标签： tensorflow python 深度学习

本文链接：https://blog.csdn.net/ftimes/article/details/126711748

版权

Note 同时被 2 个专栏收录

59 篇文章 1 订阅

订阅专栏

杂记

43 篇文章 1 订阅

订阅专栏

Tensorflow 1.x和Tensorflow 2.x读取tfrecord方法略有不同，下面分别记录两段代码：

Tensorflow 1.x：

for example in tf.python_io.tf_record_iterator(path):
    # print(tf.train.Example.FromString(example))
    jsonMessage = MessageToJson(tf.train.Example.FromString(example))

Tensorflow 2.x：

import tensorflow as tf
import json
from google.protobuf.json_format import MessageToJson

dataset = tf.data.TFRecordDataset("mydata.tfrecord")
for d in dataset:
    ex = tf.train.Example()
    ex.ParseFromString(d.numpy())
    m = json.loads(MessageToJson(ex))
    print(m['features']['feature'].keys())