tensorflow - tf.train.batch 如何创建批处理

Question

在 CIFAR-10 TensorFlow 教程中我遇到了以下行:

images, label_batch = tf.train.batch(
        [image, label],
        batch_size=batch_size,
        num_threads=num_preprocess_threads,
        capacity=min_queue_examples + 3 * batch_size)

函数tf.train.batch()似乎只将一张图像和一个标签作为输入。那么它如何创建包含多个图像的批处理？

Answer 1

它接受输入对[image, label]，是的，它是一对。然而，tf.train.batch 在内部创建了一个队列。 num_threads 线程会将这些对累积到队列中，直到达到容量。

images, label_batch 实际上是出队操作。

请记住，您正在定义一个计算图，因此[image, label]对代表该图的两个节点以及各种真实对训练集的>图像、标签将流经这些节点。这样，tf.train.batch就可以捕获图像和标签的流并填充队列。