我在使用队列时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)
请帮我
最佳答案
您可以添加文件名队列,例如string_input_producer
,以替换该文件夹中的所有文件。并尝试注释掉shuffle_batch
,看看文件名队列是否正在获取任何数据。如果您不使用num_epoch
,则此方法可能会运行多次。
关于python - OutOfRangeError(请参阅上面的回溯):FIFOQueue'_1_batch/fifo_queue'已关闭并且元素不足(请求的32,当前大小为0),我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/42224327/