我有一个使用TFRecord的tensorflow程序,想要使用tf.contrib.data.TFRecordDataset来读取数据,但是当我尝试解析示例时出现异常:“TypeError:无法将类型为< class 'xxx'>的对象转换为张量”。
仅仅使用下面这段代码时:
代码如下:
代码如下:
def _parse_function(example_proto):
features = {"var_len_feature": tf.VarLenFeature(tf.float32),
"FixedLenFeature": tf.FixedLenFeature([10], tf.int64),
"label": tf.FixedLenFeature((), tf.int32default_value=0)}
parsed_features = tf.parse_single_example(example_proto, features)
return parsed_features["image"], parsed_features["label"]
filenames = ["/var/data/file1.tfrecord", "/var/data/file2.tfrecord"]
dataset = tf.contrib.data.TFRecordDataset(filenames)
dataset = dataset.map(_parse_function)