4
我想創建一個從迭代器填充的隊列。在下面的MWE然而,總是相同的值入隊:從Python迭代器填充隊列
import tensorflow as tf
import numpy as np
# data
imgs = [np.random.randn(i,i) for i in [2,3,4,5]]
# iterate through data infinitly
def data_iterator():
while True:
for img in imgs:
yield img
it = data_iterator()
# create queue for data
q = tf.FIFOQueue(capacity=5, dtypes=[tf.float64])
# feed next element from iterator
enqueue_op = q.enqueue(list(next(it)))
# setup queue runner
numberOfThreads = 1
qr = tf.train.QueueRunner(q, [enqueue_op] * numberOfThreads)
tf.train.add_queue_runner(qr)
# dequeue
dequeue_op = q.dequeue()
dequeue_op = tf.Print(dequeue_op, data=[dequeue_op], message="dequeue()")
# We start the session as usual ...
with tf.Session() as sess:
coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(coord=coord)
for i in range(10):
data = sess.run(dequeue_op)
print(data)
.
coord.request_stop()
coord.join(threads)
難道我一定要使用feed_dict
?如果是的話,我該如何結合QueueRunner使用它?