由于有些人工智能框架会专门提供读取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

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐