如何使用tfrecord读取的数据放入到keras中训练

来源:4-8 tf.data读取tfrecord文件并与tf.keras结合使用

O_O_似水流年_O_O

2019-12-03

老师我自己尝试写一下cifar10的数据进行分类,但是卡在一个地方了
图片描述
上述是使用decode_jpeg方法产生的效果,然后送入fit之后,错误信息如下图片描述

下面是使用decode_raw方法出来的数据图片
图片描述

然后将dataset送入到fit之后就报错了,报错信息如下
图片描述

这边问下老师,一般情况直接将tfrecord读取到的dataset如何使用,需要做什么处理,还有哪两个decode什么情况下使用,deocde_raw出来的维度为什么不是32323=3072的,然后fit的使用的dataset需要怎么处理呢

写回答

1回答

正十七

2019-12-15

同学你好,对于第一个问题,维度不对是不是因为你没有做batch?

对于decode_raw的API,它是一个通用的读文件的接口,无法区分你读的文件格式,只是把字节给读进来。所以不能读成三维矩阵。而decode_jpeg可以。

https://www.tensorflow.org/api_docs/python/tf/io/decode_raw

0
1
O_O_似水流年_O_O
非常感谢!
2019-12-19
共1条回复

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程