tf.train.match_filenames_once如何验证文件是否正确读取？

原创于 2017-07-14 12:12:26 发布 · 5.1k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#python

Deep Learning 专栏收录该内容

28 篇文章

订阅专栏

本文介绍了一种简单的方法来测试TensorFlow中tfrecords文件是否成功读取。通过使用tf.train.match_filenames_once函数并结合相应的初始化操作，可以有效判断指定目录下的tfrecords文件是否被正确加载。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用到了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))

输出不是 [ ] 即代表文件已经读取到