Python读取并显示MNIST,最后保存成28*28灰度图片

重要!!!

重要!!!

重要!!!

先下载MNIST数据集,将train和test集图片解压!否则程序找不到相应文件!

注意路径!!!!!!!!!!!!!!!!!

注意路径!!!!!!!!!!!!!!!!!

注意路径!!!!!!!!!!!!!!!!!

import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
import struct
from PIL import Image
import scipy.misc


filename = r'D:\source\technology_source\data\t10k-images.idx3-ubyte'
binfile = open('C:/Users/Administrator/Desktop/python/MNIST_data/train-images.idx3-ubyte', 'rb')
buf = binfile.read()

index = 0
magic, numImages, numRows, numColumns = struct.unpack_from('>IIII', buf, index)  # 读取前4个字节的内容
index += struct.calcsize('>IIII')
for i in range(numImages):
    im = struct.unpack_from('>784B', buf, index)  # 以大端方式读取一张图上28*28=784
    index += struct.calcsize('>784B')
    im = np.array(im)
    im = im.reshape(28, 28)
#    fig = plt.figure()
#    plt.axis('off')
#    plt.imshow(im, cmap='gray')
#    plt.show()
    
    scipy.misc.imsave('C:/Users/Administrator/Desktop/python/train/'+"train" + str(i) + ".jpg", im)
    
#    im = Image.fromarray(im)
#    im.save("test" + str(i) + ".tif")
#    plt.images.imsave("test" + str(i) + ".png", im)
    plt.close()
binfile.close()

 

  • 3
    点赞
  • 22
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值