tensorflow string_input_producer 给出空队列

Question

我正在尝试根据文件名列表在 tensorflow 中创建一个队列。该列表已创建，但字符串输入生产者似乎返回了一个空队列。代码不起作用可能还有其他原因。下面是代码：

sess = tf.InteractiveSession()

def read_my_file_format(filename_queue):
    reader = tf.WholeFileReader()
    key, value = reader.read(filename_queue)
    images = tf.image.decode_jpeg(value, channels=3)
    return images, key

def input_pipeline(filenames, batch_size, num_epochs, labels):
    filename_queue = tf.train.string_input_producer(filenames, num_epochs=num_epochs, shuffle=False)
    image, key = read_my_file_format(filename_queue)
    return image, key

sess.run(tf.global_variables_initializer())

coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(sess = sess, coord = coord, start=True)

input_pipeline(trainnames, batch_size, None, labels)

coord.request_stop()
coord.join(threads)

Answer 1

如果你能提供错误信息，那会有所帮助。

同时，这里对你的代码做几点说明：

• input_pipeline 方法定义了与您的队列关联的 TensorFlow 运算符并将它们添加到图表中，因此您应该在调用 sess.run(tf.global_variables_initializer())之前调用它，如果您希望队列成为图表的一部分并且在调用 @987654325 之前@如果你想让你的队列由协调器启动。
• 您尚未要求 TensorFlow 实际运行您的 image, key 运算符。

这是我要尝试的代码：

sess = tf.InteractiveSession()

def read_my_file_format(filename_queue):
    reader = tf.WholeFileReader()
    key, value = reader.read(filename_queue)
    images = tf.image.decode_jpeg(value, channels=3)
    return images, key

def input_pipeline(filenames, batch_size, num_epochs, labels):
    filename_queue = tf.train.string_input_producer(filenames, num_epochs=num_epochs, shuffle=False)
    image, key = read_my_file_format(filename_queue)
    return image, key

# Let us define the queue operators and add them to the default graph.
image, key = input_pipeline(trainnames, batch_size, None, labels)

sess.run(tf.global_variables_initializer())

coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(sess = sess, coord = coord, start=True)

# Let's run the image, key tensors.
sess.run([image, key])

coord.request_stop()
coord.join(threads)