MINST数据集转换为png图片

原创

已于 2024-03-26 10:08:43 修改 · 426 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#机器学习 #python #深度学习

于 2020-08-28 16:49:44 首次发布

本文介绍如何将MNIST手写数字数据集转换为PNG图片格式，通过Python代码实现数据解析与图像保存，适用于机器学习初学者。

MINST数据集转换为png图片

最近在学习machine learning，希望可以出一些自己的内容，首先是相当于hello world的minst数据集处理。

MINST数据集如下：http://yann.lecun.com/exdb/mnist/

它的格式不是通用格式，需要自行调整，如下

#下面是需要导入的库
import numpy as np
import struct
from PIL import Image


def decode_idx3_ubyte(idx3_ubyte_file,saveFlag,status):

#此函数用来解析idx3文件，idx3_ubyte_filec指定图像文件路径

    #读取二进制数据
    bin_data=open(idx3_ubyte_file,'rb').read()

    #解析文件头信息，依次为魔数、图片数量、每张图片高、每张图片宽
    offest=0
    fmt_header='>iiii'
    magic_number,num_images,num_rows,num_cols=struct.unpack_from(fmt_header,bin_data,offest)
    print('魔数：%d,图片数量：%d,图片大小：%d%d' % (magic_number,num_images,num_rows,num_cols))

    #解析数据集
    image_size=num_rows