我在使用队列时读取张量流的问题有问题。请让我知道我在做什么错。以下是代码。
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)
请帮帮我
答案 0 :(得分:1)
您可以添加文件名队列(例如string_input_producer
)来运行替换文件夹中的所有文件。并尝试注释掉shuffle_batch
并查看文件名队列是否获取任何数据。如果您将num_epoch
保留为无,则此方法可以多次运行。