python - OutOfRangeError(回溯见上): FIFOQueue '_1_batch/fifo_queue' is closed and has insufficient elements (requested 32, 当前大小 0)

Question

我在使用队列时遇到 tensorflow 读取图像的问题。请让我知道我在做什么错误。下面是代码。

import tensorflow as tf
slim = tf.contrib.slim
from tensorflow.python.framework import ops

import glob
filelist = glob.glob("/*.png")
filelist[0]

imagelist = ops.convert_to_tensor(filelist)

#Makes an input queue 
input_queue = tf.train.slice_input_producer([imagelist],num_epochs = 2, shuffle = True, capacity = 64*3072)

我使用了不同的容量值，但都没有用

def read_images_from_disk(input_queue):
    file_contents = tf.read_file(input_queue[0])
    example = tf.image.decode_png(file_contents, channels=3)
    return example

image = read_images_from_disk(input_queue)

image.set_shape([28,28,3])
image_batch = tf.train.batch([image],batch_size = 32)

with tf.Session() as sess:
sess.run(tf.initialize_all_variables())
coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(coord=coord)

for i in range(20):
    print (sess.run(image_batch))

coord.request_stop()
coord.join(threads)
sess.close()


OutOfRangeError (see above for traceback): FIFOQueue '_1_batch/fifo_queue'
is closed and has insufficient elements (requested 32, current size 0)       

请帮帮我

Answer 1

您可以添加文件名队列，例如 string_input_producer，以替换文件夹中的所有文件。并尝试注释掉 shuffle_batch 并查看文件名队列是否正在获取任何数据。如果您将 num_epoch 设置为 none，此方法可能会运行多次。