在TensorFlow中将元数据与filename_queue相关联

Question

我按如下方式设置了我的阅读管道:(如this教程中所述）

filename_queue = tf.train.string_input_producer(filenames, shuffle=True)
image_reader = tf.WholeFileReader()
key, image_file = image_reader.read(filename_queue)
image = tf.image.decode_png(image_file, channels)
# need to access the file name here!

如何在阅读图像时访问文件名（或我可以与文件关联的其他标识符）？

Answer 1

通常，使用当前的TensorFlow管道实现这一点很棘手，但幸运的是tf.WholeFileReader使其变得简单：key张量（由tf.WholeFileReader.read()返回）包含文件名数据存储在image_file张量中的文件。您可以在任何想要访问文件名的地方使用key。