由于有些人工智能框架会专门提供读取MNIST数据集的API(mdsp。。。),所以有时候想着可以把jpg或者png自己转成MNIST格式。

事实上,MNIST的数据格式也不复杂,硬写也是可以写的,这里主要介绍一个github上的大神的代码

一、使用步骤

源码在https://github.com/gskielian/JPG-PNG-to-MNIST-NN-Format
目录结构为
在这里插入图片描述
项目只支持在Linux上运行,但是不用害怕,我们可以用虚拟机!
官方环境python2 + pillow
不过python3一样可以跑(那个imagemagick什么的可以不用装)
batches.meta.txt文件官方说是放标签,但是我实验了好像不是,真正的标签是两个图片文件夹下的文件夹名字,例如
在这里插入图片描述
则0文件夹下放0的图片,标签也就是0
然后默认是png格式。要用jpg格式就打开convert-images-to-mnist-format.py把里面的png全部换成jpg就可以了。
最后就是跑一下convert-images-to-mnist-format.py就🆗了

最重要的是图片必须是二值图(黑白),困扰我3小时。。。

Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐