MNIST数据集格式转化

最新推荐文章于 2024-02-27 21:40:43 发布

jiarenyf

最新推荐文章于 2024-02-27 21:40:43 发布

阅读量4.4k

点赞数 2

分类专栏： machine-learning caffe python

本文链接：https://blog.csdn.net/u011762313/article/details/50506596

版权

MNIST数据集是ubyte格式存储的，现在转化为png格式：
训练集：

import numpy as np
import struct

from PIL import Image
import os

data_file = 'train-images-idx3-ubyte'
# It's 47040016B, but we should set to 47040000B
data_file_size = 47040016
data_file_size = str(data_file_size - 16) + 'B'

data_buf = open(data_file, 'rb').read()

magic, numImages, numRows, numColumns = struct.unpack_from(
    '>IIII', data_buf, 0)
datas = struct.unpack_from(
    '>' + data_file_size, data_buf, struct.calcsize('>IIII'))
datas = np.array(datas).astype(np.uint8).reshape(
    numImages,