OutOfRangeError(参见上面的回溯):FIFOQueue' _1_batch / fifo_queue'关闭且元素不足(请求32,当前大小0)

时间:2017-02-14 10:55:50

标签: python tensorflow queue

我在使用队列时读取张量流的问题有问题。请让我知道我在做什么错。以下是代码。

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)       

请帮帮我

1 个答案:

答案 0 :(得分:1)

您可以添加文件名队列(例如string_input_producer)来运行替换文件夹中的所有文件。并尝试注释掉shuffle_batch并查看文件名队列是否获取任何数据。如果您将num_epoch保留为无,则此方法可以多次运行。