使用tf.data.TFRecordDataset读取tfrecord中的图片数据并显示

最新推荐文章于 2022-04-03 23:09:29 发布

LukeDing80

最新推荐文章于 2022-04-03 23:09:29 发布

阅读量1.1k

点赞数

文章标签： tensorflow

本文链接：https://blog.csdn.net/qq_20414353/article/details/119534466

版权

本文介绍了一种在TensorFlow 2.5和Python 3.8环境下，利用tf.data.TFRecordDataset高效读取TFRecord文件中的图片数据并进行显示的方法。适用于需要快速访问TFRecord中存储的图像数据的场景。

摘要由CSDN通过智能技术生成

读取tfrecord文件的方法有很多种，如果只是想单纯快速的读取出tfrecord文件中的图片，并显示它，那么下面这个方法非常的简洁高效。
我的tensorflow是2.5版的，python是3.8版本

import tensorflow as tf
import numpy as np
import IPython.display as display
#tfrecord文件路径根据自己的修改
filenames = ['D:/cats_dogs/annotations/eval.record']
raw_image_dataset = tf.data.TFRecordDataset(filenames)


# 创建特征字典，使用TF官方教程创建的标准tfrecord文件，特征标签和下面是一致的
image_feature_description = {
   'image/encoded':tf.io.FixedLenFeature([]<