一起来用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