用到了tf.train.match_filenames_once这个函数去读取tfrecords文件,却不知道文件是否已经读取到,用以下这段代码可以轻松测试。
import tensorflow as tf
directory = "*.*"
file_names = tf.train.match_filenames_once(directory)
init = (tf.global_variables_initializer(), tf.local_variables_initializer())
with tf.Session() as sess:
sess.run(init)
print(sess.run(file_names))
输出不是 [ ] 即代表文件已经读取到