浣熊

一起来用tf.data API!(4)——使用图像数据制作TFRecords文件

蹲街弑〆低调 提交于 2019-12-20 19:36:13
一起来用tf.data API!(4)——使用图像数据制作TFRecords文件 (一)前 言 (二)图片收集归类 (三)TFRecords文件生成 (四)总 结 (一)前 言 在上一节中我们对TFRecords做了简要的介绍,这一节我们来演示如何使用图像数据制作TFRecords。本次我们制作一个含有浣熊、大熊猫两类动物的数据集。 (二)图片收集归类 收集若干浣熊、大熊猫的图片,如下图所示: 在同一目录下创建两个文件夹,分别命名为XM(熊猫)、HX(浣熊),如下图所示: 然后将图片分别放入这两个文件中,开始进行下一步。 (三)TFRecords文件生成 运行上述源码,在你指定的输出位置发现已经生成了TFRecords文件,如图所示: (四)总 结 在本节中我们介绍了如何将图像数据制作为TFRecords文件,在下一节中我们将使用tf.data API读取生成的TFRecords,并将其重新显示,有任何的疑问可以在评论区留言,我会尽快回复,谢谢支持! 来源: CSDN 作者: like_study_cat 链接: https://blog.csdn.net/like_study_cat/article/details/103636651