关联与filename_queue元数据TensorFlow

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。