如何解析单个TFRecord?

时间:2019-04-04 04:12:04

标签: tensorflow

我有以下代码:

def bytes_feature(value):
  return tf.train.Feature(bytes_list=tf.train.BytesList(value=[value]))

tf_example = tf.train.Example(features=tf.train.Features(feature={
    'image/encoded': bytes_feature(encoded_image_data),
}))

我现在正在尝试通过以下方式解析TFRecord:

features = tf.parse_single_example(
    tf_example, 
    features={'image/encoded': tf.FixedLenFeature((), tf.string)}
)

但是,我不断收到错误消息: Failed to convert object of type <class 'tensorflow.core.example.example_pb2.Example'> to Tensor.

我在做什么错了?

0 个答案:

没有答案