關聯與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。