一起来用tf.data API！（4）——使用图像数据制作TFRecords文件

一起来用tf.data API！（4）——使用图像数据制作TFRecords文件

（一）前 言
（二）图片收集归类
（三）TFRecords文件生成
（四）总 结

（一）前 言

在上一节中我们对TFRecords做了简要的介绍，这一节我们来演示如何使用图像数据制作TFRecords。本次我们制作一个含有浣熊、大熊猫两类动物的数据集。

（二）图片收集归类

收集若干浣熊、大熊猫的图片，如下图所示：

在同一目录下创建两个文件夹，分别命名为XM（熊猫）、HX（浣熊），如下图所示：

然后将图片分别放入这两个文件中，开始进行下一步。

（三）TFRecords文件生成

运行上述源码，在你指定的输出位置发现已经生成了TFRecords文件，如图所示：

（四）总 结

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