tensorflow数据读取之文件读取流程

最新推荐文章于 2024-04-11 10:00:00 发布

写bug战神

最新推荐文章于 2024-04-11 10:00:00 发布

阅读量295

点赞数 1

分类专栏： tensorflow 文章标签： tensorflow 文件读取

本文链接：https://blog.csdn.net/qq_40064490/article/details/102531959

版权

tensorflow 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

在这里插入图片描述

这里的string_tensor直接写入“文件名、路径”所构成的列表即可，tensorflow自动将其转换为一阶张量；shuffle=True作用为打乱传入的文件名的顺序。

# 1、构造一个文件名队列
file_queue = tf.train.string_input_producer(string_tensor,shuffle=True)

在这里插入图片描述

# 2、读取与解码
#    先实例化读取器与解码器，举例如下：
# 文本文件
file_reader = tf.TextLineReader()
# 图片文件
file_reader = tf.WholeFileReader()
# 二进制文件
file_reader = tf.FixedLengthRecordReader(record_bytes)
# TFRecords文件
file_reader = tf.TFRecordReader()

#     之后用实例化的文件读取器调用read(file_queue)方法进行文件读取
key, value = file_reader.read(file_queue)
# 文本文件解码
tf.decode_csv()
# 图片文件解码
tf.image.decode_jpeg(contents)
tf.image.decode_png(contents)
# 二进制文件解码
tf.decode_raw()