【python】将普通图像(PNG)转换为MNIST数据集格式(idx1-ubyte)

源代码:https://github.com/gskielian/JPG-PNG-to-MNIST-NN-Format

网络上关于深度学习的源实例代码都是基于MNIST手写字体数据集的,该数据集格式比较特殊,若想应用于自己的数据应该怎么转换呢?

在github上找到了大神编的代码,整个文件如下:

原始图像分别放到train-images和test-images文件夹中,注意默认格式为PNG,若是JPG要修改代码,而且图像要按照标签存放,数字标签要从1开始

同时,batches.meta.txt存放对应标签。

resize-script.sh文件用于改变图像尺寸,统一为28*28

修改convert-image-to-mnist-format.py图像文件路径即可成功实现图像格式的转换,生成

 

posted @ 2018-08-16 15:08  奥古斯都乀  阅读(11966)  评论(21编辑  收藏  举报