0
点赞
收藏
分享

微信扫一扫

Tensorflow 入门学习15.下载MNIST手写体数据集及保存图片

Tensorflow 入门学习15.下载MNIST手写体数据集及保存图片

下载数据集

#!/usr/bin/python
# -*- coding: UTF-8 -*-

import input_data

# 60000行的训练数据集(mnist.train)和10000行的测试数据集(mnist.test)
mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)

将数据集保存为图片

import struct
import numpy as np
import PIL.Image

filename='./MNIST_data/train-images.idx3-ubyte'
binfile=open(filename,'rb')
buf=binfile.read()
index=0
magic,numImages,numRows,numColumns=struct.unpack_from('>IIII',buf,index)
index+=struct.calcsize('>IIII')
for image in range(0,numImages):
im=struct.unpack_from('>784B',buf,index)
index+=struct.calcsize('>784B')
im=np.array(im,dtype='uint8')
im=im.reshape(28,28)
im=PIL.Image.fromarray(im)
im.save('mnist_train/train_%s.bmp'%image,'bmp')

参考:​


举报

相关推荐

0 条评论