TensorFlow中有三种方法可以用来读取数据,第一种是直接使用constant给常量赋值;第二种是使用dict进行feed,但是速度会很慢,因为需要先把文件整个读到内存中。
第三种也是最常用的是使用TF中自带的datareaders来进行:TF 包含了四种DataReaders:
tf.TextLineReader
最常用的,可以按行读取文件。
tf.FixedLengthRecordReader
可以输出所有整个文件的所有数据(要求所有数据长度,格式相同)
tf.WholeFIleReader
直接输出整个文件(一般用于一个文件就是一组数据的情况)
tf.TFRecordReader
读取TF自身的二进制格式文件(TFRecord)
tf.ReaderBase
可以作为基类来定制自己的Reader.
时间: 2024-09-30 21:48:56