python如何读取tfrecord_如何将tfrecord数据读入tensors/numpy数组?

我有一个tfrecord文件,其中存储了一个数据列表,每个元素都有二维坐标和三维坐标。坐标是dtype float64的2d numpy数组。在

这些是我用来存储它们的功能。在feature = {'train/coord2d': _floats_feature(projC),

'train/coord3d': _floats_feature(sChair)}

我把它们放在一个浮动列表中。在

^{pr2}$

现在我正在尝试接收它们,以便将它们输入到我的网络中进行训练。我希望2d坐标作为输入,3d作为输出,用于训练我的网络。在def read_and_decode(filename):

filename_queue = tf.train.string_input_producer(filename, name='queue')

reader = tf.TFRecordReader()

_, serialized_example = reader.read(filename_queue)

features = tf.parse_single_example(

serialized_example,

features= {'train/coord2d': tf.FixedLenFeature([], tf.float32),

'train/coord3d': tf.FixedLenFeature([], tf.float32)})

coord2d = tf.cast(features['train/coord2d'], tf.float32)

coord3d = tf.cast(features['train/coord3d'], tf.float32)

return coord2d, coord3d

with tf.Session() as sess:

filename = ["train.tfrecords"]

dataset = tf.data.TFRecordDataset(filename)

c2d, c3d = read_and_decode(filename)

print(sess.run(c2d))

print(sess.run(c3d))

这是我的代码,但我并不真正理解它,因为我从教程等,所以我试图打印出c2d和c3d,看看它们是什么格式的,但我的程序一直在运行,根本没有打印任何东西,而且从未终止。c2d和c3d是否包含数据集中每个元素的2d和3d坐标?当训练网络作为输入和输出时,它们能直接使用吗?在

我也不知道他们应该是什么格式之前,他们可以作为输入到网络。我应该把它们转换回2d numpy数组还是2d张量?万一我怎么办?总的来说,我只是非常失落,所以任何一个圭达奇将是非常有帮助的!谢谢

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值