tensorflow获取string_input_producer的大小

Question

如何检查string_input_producer已读取的文件名数？将根据输入数据的大小执行不同的操作，因此我需要知道将读取或已读取的图像数量。

以下代码没有告诉我我已阅读或即将阅读的图片数量。

{{1}}

Answer 1

像string_input_producer这样的函数会向当前图形添加一个队列，每次只能只出列一个 。通常输出张量将输入到tf.train.shuffle_batch这样的函数，这是你想要的。此函数的参数batch_size可以控制每次作为模型输入出列的示例数

更新：

如果要检查输入数据是否正确，可以使用sess.run(my_img)运行它，这会给你一个numpy.array张量。您可以直接查看此张量的元素，或者只使用matplotlib绘制它。

确保您已在sess.run之前启动了队列运行器，否则您的程序将永久挂起

Answer 2

string_input_producer会返回一个标准FIFOQueue（它会返回input_producer，然后会返回queue。

FIFOQueue没有关于它已读取的元素数量的信息，只有当前在队列中的元素数量（q.size()）。如果您想知道已经读取了多少元素，您需要手动添加一个计数器，每次读取元素时都会递增。