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

最新推荐文章于 2021-01-28 22:18:07 发布

Glory@OUC

最新推荐文章于 2021-01-28 22:18:07 发布

阅读量5.1k

点赞数 3

分类专栏： Deep Learning 文章标签： python

本文链接：https://blog.youkuaiyun.com/zsg2063/article/details/75103654

版权

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))

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