一起来用tf.data API！（5）——使用tf.data API读取TFRecords文件

一起来用tf.data API！（5）——使用tf.data API读取TFRecords文件

（一）前 言
（二）使用tf.data API读取文件
    （1）定义数据预处理操作
    （2）创建dataset
    （3）创建迭代器
    （4）定义会话取出数据
    （5）完整代码
（三）总 结

（一）前 言

在上一节中，我们成功将图像数据制作成了TFRecords文件，在这一节中我们要使用tf.data API将其读取出来，并使用matplotlib对其进行显示。

（二）使用tf.data API读取文件

我们通过如下的代码实现这一操作：

（1）定义数据预处理操作

注意在进行tfrecords读取的时候，还原特征列的属性一定要与写入时创建的example相同

（2）创建dataset

（3）创建迭代器

（4）定义会话取出数据

（5）完整代码

运行上述源码，显示如下：

（三）总 结

在本节中我们介绍了如何使用tf.data API读取生成的TFRecords，并将其重新显示，有任何的疑问可以在评论区留言，我会尽快回复，谢谢支持！