如何使用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回答
-
同学你好,对于第一个问题,维度不对是不是因为你没有做batch?
对于decode_raw的API,它是一个通用的读文件的接口,无法区分你读的文件格式,只是把字节给读进来。所以不能读成三维矩阵。而decode_jpeg可以。
https://www.tensorflow.org/api_docs/python/tf/io/decode_raw
012019-12-19
相似问题