用到了tf.train.match_filenames_once这个函数去读取tfrecords文件,却不知道文件是否已经读取到,用以下这段代码可以轻松测试。
import tensorflow as tf
directory = "*.*"
file_names = tf.train.match_filenames_once(directory)
init = ( tf.local_variables_initializer())
with tf.Session() as sess:
sess.run(init)
print(sess.run(file_names))
注意:输出不是 [ ] 即代表文件已经读取到
name = #开始没有读到
[’…‘] #出现路径表示读到了
解决方法: 因为此例只有一个局部变量,所以把sess.run(tf.global_variables_initializer())换成sess.run(tf.local_variables_initializer())即可
以下为参考的内容:
import tensorflow as tf
#使用tf.train.match_filenames_once函数获取文件列表
files=tf.train.match_filenames_once(“data.tfrecords-*”)
#通过tf.train.string_input_producer函数创建输入队列,输入队列中的文件列表为
#tf.train.mach_filenames_once函数获取的文件列表。这里将shuffle参数设