模仿mnist数据集制作自己的数据集
最近看深度学习,用TensorFlow跑教程里的模型,发现教程里大多都是用mnist和cifar这两个数据集来演示的。想测试自己的数据集看看初步效果,于是就想套用现有的模型,将自己的数据集做成和mnist或cifar数据集格式一模一样的格式。然后就发现Tensorflow里没有现成的方法可以将图片转换成类似于mnist和cifar数据集格式的方法。于是就百度,发现网上几乎同样没有这样的代码,于是就各种查资料,最后自己终于实现了,也是心累啊~~~
好了废话不多说,直接进入主题:
1、关于mnist数据集的介绍,可以去官网查看mnist数据集,也可以看我的另一篇文章(超详细)mnist数据集介绍、读取、保存成图片
这里就不详细介绍了,直接上代码:
代码一共分三个文件:
Image_to_BinaryData.h
Image_to_BinaryData.cpp
Main.cpp
其中:
Image_to_BinaryData.h和Image_to_BinaryData.cpp里面封装了相关的