当前位置: 开发笔记 > 编程语言 > 正文

TensorFlow数据读取的四种方法总结

作者：jack_wangzhu | 来源：互联网 | 2023-09-03 14:03

读取数据（Readingdata）推荐：如何构建高性能的输入pipelineTensorFlow输入数据的方法有四种：

读取数据&＃xff08;Reading data&＃xff09;
&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;推荐&＃xff1a;如何构建高性能的输入 pipeline&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;
TensorFlow输入数据的方法有四种&＃xff1a;
1、tf.data API&＃xff1a;可以很容易的构建一个复杂的输入通道(pipeline)&＃xff08;首选数据输入方式&＃xff09;&＃xff08;Eager模式必须使用该API来构建输入通道&＃xff09;
2、Feeding&＃xff1a;使用Python代码提供数据&＃xff0c;然后将数据feeding到计算图中。
3、QueueRunner&＃xff1a;基于队列的输入通道&＃xff08;在计算图计算前从队列中读取数据&＃xff09;
4、Preloaded data&＃xff1a;用一个constant常量将数据集加载到计算图中&＃xff08;主要用于小数据集&＃xff09;

目录
读取数据&＃xff08;Reading data&＃xff09;
1. tf.data API
2. Feeding
3. QueueRunner
3.1 Filenames, shuffling, and epoch limits
3.2 File formats
3.2.1 CSV file
3.2.2 Fixed length records
3.2.3 Standard TensorFlow format
3.3 Preprocessing
3.4 Batching
3.5 Creating threads to prefetch using QueueRunner objects
3.6 Filtering records or producing multiple examples per record
3.7 Sparse input data
4. Preloaded data
5. Multiple input pipelines

1. tf.data API
关于tf.data.Dataset的更详尽解释请看《programmer’s guide》。tf.data API能够从不同的输入或文件格式中读取、预处理数据&＃xff0c;并且对数据应用一些变换&＃xff08;例如&＃xff0c;batching、shuffling、mapping function over the dataset&＃xff09;&＃xff0c;tf.data API 是旧的 feeding、QueueRunner的升级。
tf.data的教程见下篇: TensorFlow导入数据&＃xff08;tf.data&＃xff09;

2. Feeding
注意&＃xff1a;Feeding是数据输入效率最低的方式&＃xff0c;应该只用于小数据集和调试&＃xff08;debugging&＃xff09;
TensorFlow的Feeding机制允许我们将数据输入计算图中的任何一个Tensor。因此可以用Python来处理数据&＃xff0c;然后直接将处理好的数据feed到计算图中。
在run()或eval()中用feed_dict来将数据输入计算图&＃xff1a;
`with tf.Session():input &＃61; tf.placeholder(tf.float32)classifier &＃61; ...print(classifier.eval(feed_dict&＃61;{input: my_python_preprocessing_fn()}))`
虽然你可以用feed data替换任何Tensor的值&＃xff08;包括variables和constants&＃xff09;&＃xff0c;但最好的使用方法是使用一个tf.placeholder节点&＃xff08;专门用于feed数据&＃xff09;。它不用初始化&＃xff0c;也不包含数据。一个placeholder没有被feed数据&＃xff0c;则会报错。
使用placeholder和feed_dict的一个实例(数据集使用的是MNIST)见tensorflow/examples/tutorials/mnist/fully_connected_feed.py

3. QueueRunner
注意&＃xff1a;这一部分介绍了基于队列&＃xff08;Queue&＃xff09;API构建输入通道&＃xff08;pipelines&＃xff09;&＃xff0c;这一方法完全可以使用 tf.data API来替代。
一个基于queue的从文件中读取records的通道&＃xff08;pipline&＃xff09;一般有以下几个步骤&＃xff1a;
1、文件名列表&＃xff08;The list of filenames&＃xff09;
2、文件名打乱&＃xff08;可选&＃xff09;&＃xff08;Optional filename shuffling&＃xff09;
3、epoch限制&＃xff08;可选&＃xff09;&＃xff08;Optional epoch limit&＃xff09;
4、文件名队列&＃xff08;Filename queue&＃xff09;
5、与文件格式匹配的Reader&＃xff08;A Reader for the file format&＃xff09;
6、decoder&＃xff08;A decoder for a record read by the reader&＃xff09;
7、预处理&＃xff08;可选&＃xff09;&＃xff08;Optional preprocessing&＃xff09;
8、Example队列&＃xff08;Example queue&＃xff09;

3.1 Filenames, shuffling, and epoch limits

对于文件名列表&＃xff0c;有很多方法&＃xff1a;

1. 使用一个constant string Tensor&＃xff08;比如&＃xff1a;["file0", "file1"]&＃xff09;或者[("file%d" %i) for i in range(2)]&＃xff1b;

2. 使用 tf.train.match_filenames_once 函数&＃xff1b;3. 使用 tf.gfile.Glob(path_pattern)。

将文件名列表传给 tf.train.string_input_producer 函数。string_input_producer 创建一个 FIFO 队列来保存&＃xff08;holding&＃xff09;文件名&＃xff0c;以供Reader使用。

string_input_producer 可以对文件名进行shuffle&＃xff08;可选&＃xff09;、设置一个最大迭代 epochs 数。在每个epoch&＃xff0c;一个queue runner将整个文件名列表添加到queue&＃xff0c;如果shuffle&＃61;True&＃xff0c;则添加时进行shuffle。This procedure provides a uniform sampling of files, so that examples are not under- or over- sampled relative to each other。

queue runner线程独立于reader线程&＃xff0c;所以enqueuing和shuffle不会阻碍reader。

3.2 File formats

要选择与输入文件的格式匹配的reader&＃xff0c;并且要将文件名队列传递给reader的 read 方法。read 方法输出一个 key identifying the file and record&＃xff08;在调试过程中非常有用&＃xff0c;如果你有一些奇怪的 record&＃xff09;

3.2.1 CSV file

为了读取逗号分隔符分割的text文件&＃xff08;csv&＃xff09;&＃xff0c;要使用一个 tf.TextLineReader 和一个 tf.decode_csv。例如&＃xff1a;

filename_queue &＃61; tf.train.string_input_producer(["file0.csv", "file1.csv"])reader &＃61; tf.TextLineReader() key, value &＃61; reader.read(filename_queue)# Default values, in case of empty columns. Also specifies the type of the # decoded result. record_defaults &＃61; [[1], [1], [1], [1], [1]] col1, col2, col3, col4, col5 &＃61; tf.decode_csv(value, record_defaults&＃61;record_defaults) features &＃61; tf.stack([col1, col2, col3, col4])with tf.Session() as sess:# Start populating the filename queue.coord &＃61; tf.train.Coordinator()threads &＃61; tf.train.start_queue_runners(coord&＃61;coord)for i in range(1200):# Retrieve a single instance:example, label &＃61; sess.run([features, col5])coord.request_stop()coord.join(threads)

read 方法每执行一次&＃xff0c;会从文件中读取一行。然后 decode_csv 将读取的内容解析成一个Tensor列表。参数 record_defaults 决定解析产生的Tensor的类型&＃xff0c;另外&＃xff0c;如果输入中有缺失值&＃xff0c;则用record_defaults 指定的默认值来填充。

在使用run或者eval 执行 read 方法前&＃xff0c;你必须调用 tf.train.start_queue_runners 去填充 queue。否则&＃xff0c;read 方法将会堵塞&＃xff08;等待 filenames queue 中 enqueue 文件名&＃xff09;。

3.2.2 Fixed length records

为了读取二进制文件&＃xff08;二进制文件中&＃xff0c;每一个record都占固定bytes&＃xff09;&＃xff0c;需要使用一个 tf.FixedLengthRecordReader 和 tf.decode_raw。decode_raw 将 reader 读取的 string 解析成一个uint8 tensor。

例如&＃xff0c;二进制格式的CIFAR-10数据集中的每一个record都占固定bytes&＃xff1a;label占1 bytes&＃xff0c;然后后面的image数据占3072 bytes。当你有一个unit8 tensor时&＃xff0c;通过切片便可以得到各部分并reformat成需要的格式。对于CIFAR-10数据集的reading和decoding&＃xff0c;可以参照&＃xff1a;tensorflow_models/tutorials/image/cifar10/cifar10_input.py或这个教程。

3.2.3 Standard TensorFlow format

另一个方法是将数据集转换为一个支持的格式。这个方法使得数据集和网络的混合和匹配变得简单&＃xff08;make it easier to mix and match data sets and network architectures&＃xff09;。TensorFlow中推荐的格式是 TFRecords文件&＃xff0c;TFRecords中包含 tf.train.Example protocol buffers (在这个协议下&＃xff0c;特征是一个字段).
你写一小段程序来获取数据&＃xff0c;然后将数据填入一个Example protocol buffer&＃xff0c;并将这个 protocol buffer 序列化&＃xff08;serializes&＃xff09;为一个string&＃xff0c;然后用 tf.python_io.TFRcordWriter 将这个string写入到一个TFRecords文件中。例如&＃xff0c;tensorflow/examples/how_tos/reading_data/convert_to_records.py 将MNIST数据集转化为TFRecord格式。

读取TFRecord文件的推荐方式是使用 tf.data.TFRecordDataset&＃xff0c;像这个例子一样:

dataset &＃61; tf.data.TFRecordDataset(filename) dataset &＃61; dataset.repeat(num_epochs)# map takes a python function and applies it to every sample dataset &＃61; dataset.map(decode)

为了完成相同的任务&＃xff0c;基于queue的输入通道需要下面的代码&＃xff08;使用的decode和上一段代码一样&＃xff09;&＃xff1a;

filename_queue &＃61; tf.train.string_input_producer([filename], num_epochs&＃61;num_epochs) reader &＃61; tf.TFRecordReader() _, serialized_example &＃61; reader.read(filename_queue) image,label &＃61; decode(serialized_example)

3.3 Preprocessing

然后你可以对examples进行你想要的预处理&＃xff08;preprocessing&＃xff09;。预处理是独立的&＃xff08;不依赖于模型参数&＃xff09;。常见的预处理有&＃xff1a;数据的标准化&＃xff08;normalization of your data&＃xff09;、挑选一个随机的切片&＃xff0c;添加噪声&＃xff08;noise&＃xff09;或者畸变&＃xff08;distortions&＃xff09;等。具体的例子见&＃xff1a;tensorflow_models/tutorials/image/cifar10/cifar10_input.py

3.4 Batching

在pipeline的末端&＃xff0c;我们通过调用 tf.train.shuffle_batch 来创建两个queue&＃xff0c;一个将example batch起来 for training、evaluation、inference&＃xff1b;另一个来shuffle examples的顺序。

例子&＃xff1a;def read_my_file_format(filename_queue):

reader &＃61; tf.SomeReader()key, record_string &＃61; reader.read(filename_queue)example, label &＃61; tf.some_decoder(record_string)processed_example &＃61; some_processing(example)return processed_example, labeldef input_pipeline(filenames, batch_size, num_epochs&＃61;None):filename_queue &＃61; tf.train.string_input_producer(filenames, num_epochs&＃61;num_epochs, shuffle&＃61;True)example, label &＃61; read_my_file_format(filename_queue)# min_after_dequeue defines how big a buffer we will randomly sample# from -- bigger means better shuffling but slower start up and more# memory used.# capacity must be larger than min_after_dequeue and the amount larger# determines the maximum we will prefetch. Recommendation:# min_after_dequeue &＃43; (num_threads &＃43; a small safety margin) * batch_sizemin_after_dequeue &＃61; 10000capacity &＃61; min_after_dequeue &＃43; 3 * batch_sizeexample_batch, label_batch &＃61; tf.train.shuffle_batch([example, label], batch_size&＃61;batch_size, capacity&＃61;capacity,min_after_dequeue&＃61;min_after_dequeue)return example_batch, label_batch

如果你需要更多的并行或者打乱不同文件中example&＃xff0c;使用多个reader&＃xff0c;然后使用 tf.train.shuffle_batch_join将多个reader读取的内容整合到一起。&＃xff08;If you need more parallelism or shuffling of examples between files, use multiple reader instances using the tf.train.shuffle_batch_join&＃xff09;

例子&＃xff1a;

def read_my_file_format(filename_queue):reader &＃61; tf.SomeReader()key, record_string &＃61; reader.read(filename_queue)example, label &＃61; tf.some_decoder(record_string)processed_example &＃61; some_processing(example)return processed_example, labeldef input_pipeline(filenames, batch_size, read_threads, num_epochs&＃61;None):filename_queue &＃61; tf.train.string_input_producer(filenames, num_epochs&＃61;num_epochs, shuffle&＃61;True)example_list &＃61; [read_my_file_format(filename_queue)for _ in range(read_threads)]min_after_dequeue &＃61; 10000capacity &＃61; min_after_dequeue &＃43; 3 * batch_sizeexample_batch, label_batch &＃61; tf.train.shuffle_batch_join(example_list, batch_size&＃61;batch_size, capacity&＃61;capacity,min_after_dequeue&＃61;min_after_dequeue)return example_batch, label_batch

所有的reader共享一个filename queue。这种方式保证了不同的reader在同一个epoch&＃xff0c;读取不同的文件&＃xff0c;直到所有的文件的已经读取完&＃xff0c;然后在下一个epoch&＃xff0c;重新从所有的文件读取&＃xff08;You still only use a single filename queue that is shared by all the readers. That way we ensure that the different readers use different files from the same epoch until all the files from the epoch have been started. (It is also usually sufficient to have a single thread filling the filename queue.)&＃xff09;。

另一个可选的方法是去通过调用 tf.train.shuffle_batch 使用单个的reader&＃xff0c;但是将参数 num_threads 参数设置为大于1的值。这将使得在同一时间只能从一个文件读取内容&＃xff08;但是比 1 线程快&＃xff09;&＃xff0c;而不是同时从N个文件中读取。这可能很重要&＃xff1a;

如果你的num_threads参数值比文件的数量多&＃xff0c;那么很有可能&＃xff1a;有两个threads会一前一后从同一个文件中读取相同的example。这是不好的&＃xff0c;应该避免。
或者&＃xff0c;如果并行地读取N个文件&＃xff0c;可能或导致大量的磁盘搜索&＃xff08;意思是&＃xff0c;多个文件存在于磁盘的不同位置&＃xff0c;而磁头只能有一个位置&＃xff0c;所以会增加磁盘负担&＃xff09;
那么需要多少个线程呢&＃xff1f;tf.train.shuffle_batch*函数会给计算图添加一个summary来记录 example queue 的使用情况。如果你有足够的reading threads&＃xff0c;这个summary将会总大于0。你可以用TensorBoard来查看训练过程中的summaries

3.5 Creating threads to prefetch using QueueRunner objects

使用QueueRunner对象来创建threads来prefetch数据

说明&＃xff1a;tf.train里的很多函数会添加tf.train.QueueRunner对象到你的graph。这些对象需要你在训练或者推理前&＃xff0c;调用tf.train.start_queue_runners&＃xff0c;否则数据无法读取到图中。调用tf.train.start_queue_runners会运行输入pipeline需要的线程&＃xff0c;这些线程将example enqueue到队列中&＃xff0c;然后dequeue操作才能成功。这最好和tf.train.Coordinator配合着用&＃xff0c;当有错误时&＃xff0c;它会完全关闭掉开启的threads。如果你在创建pipline时设置了迭代epoch数限制&＃xff0c;将会创建一个epoch counter的局部变量&＃xff08;需要初始化&＃xff09;。下面是推荐的代码使用模板&＃xff1a;

# Create the graph, etc. init_op &＃61; tf.global_variables_initializer()# Create a session for running operations in the Graph. sess &＃61; tf.Session()# Initialize the variables (like the epoch counter). sess.run(init_op)# Start input enqueue threads. coord &＃61; tf.train.Coordinator() threads &＃61; tf.train.start_queue_runners(sess&＃61;sess, coord&＃61;coord)try:while not coord.should_stop():# Run training steps or whateversess.run(train_op)except tf.errors.OutOfRangeError:print(&＃39;Done training -- epoch limit reached&＃39;) finally:# When done, ask the threads to stop.coord.request_stop()# Wait for threads to finish. coord.join(threads) sess.close()

这里的代码是怎么工作的&＃xff1f;
首先&＃xff0c;我们创建整个图。它的input pipeline将有几个阶段&＃xff0c;这些阶段通过Queue连在一起。第一个阶段&＃xff1a;将会产生要读取的文件的文件名&＃xff0c;并将文件名enqueue到filename queue。第二个阶段&＃xff1a;使用一个Reader来dequeue文件名并读取&＃xff0c;产生example&＃xff0c;并将example enqueue到一个example queue。根据你的设置&＃xff0c;你可能有很多第二阶段&＃xff08;并行&＃xff09;&＃xff0c;所以你可以从并行地读取多个文件。最后一个阶段是一个enqueue操作&＃xff0c;将example enqueue成一个queue&＃xff0c;然后等待下一步操作。我们想要开启多个线程运行着些enqueue操作&＃xff0c;所以我们的训练loop能够从example queue中dequeue examples。

è¿éåå¾çæè¿°

tf.train里的辅助函数&＃xff08;创建了这些queue、enqueuing操作&＃xff09;会调用tf.train.add_queue——runner添加一个tf.train.QueueRunner到图中。每一个QueueRunner负责一个阶段。一旦图构建好&＃xff0c;tf.train.start_queue_runners函数会开始图中每一个QueueRunner的入队操作。

如果一切进行顺利&＃xff0c;你现在可以运行训练step&＃xff08;后台线程会填满queue&＃xff09;。如果你设置了epoch限制&＃xff0c;在达到固定的epoch时&＃xff0c;在进行dequeuing会得到tf.errors.OutOfRangeError。这个错误等价于EOF&＃xff08;end of file&＃xff09;&＃xff0c;意味着已经达到了固定的epochs。

最后一部分是tf.train.Coordinator。它主要负责通知所有的线程是否应该停止。在大多数情况下&＃xff0c;这通常是因为遇到了一个异常&＃xff08;exception&＃xff09;。例如&＃xff0c;某一个线程在运行某些操作时出错了&＃xff08;或者python的异常&＃xff09;。

关于threading、queues、QueueRunners、Coordinators的更多细节见这里

3.6 Filtering records or producing multiple examples per record

一个example的shape是 [x,y,z]&＃xff0c;一个batch的example的shape为 [batch, x, y, z]。如果你想去过滤掉这个record&＃xff0c;你可以把 batch size 设置为 0&＃xff1b;如果你想让每一个record产生多个example&＃xff0c;你可以把batch size设置为大于1。然后&＃xff0c;在调用调用batching函数&＃xff08;shuffle_batch或shuffle_batch_join&＃xff09;时&＃xff0c;设置enqueue_many&＃61;True。

3.7 Sparse input data

queues在SparseTensors的情况下不能很好的工作。如果你使用SparseTensors&＃xff0c;你必须在batching后用tf.sparse_example来decode string records&＃xff08;而不是在batching前使用tf.parse_single_example来decode&＃xff09;