MNIST数据集转化为二维图片的实现示例

yipeiwu_com5年前Python基础

本文介绍了MNIST数据集转化为二维图片的实现示例,分享给大家,具体如下:

#coding: utf-8
from tensorflow.examples.tutorials.mnist import input_data
import scipy.misc
import os

# 读取MNIST数据集。如果不存在会事先下载。
mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)

# 我们把原始图片保存在MNIST_data/raw/文件夹下
# 如果没有这个文件夹会自动创建
save_dir = 'MNIST_data/raw/'
if os.path.exists(save_dir) is False:
  os.makedirs(save_dir)

# 保存前20张图片
for i in range(20):
  # 请注意,mnist.train.images[i, :]就表示第i张图片(序号从0开始)
  image_array = mnist.train.images[i, :]
  # TensorFlow中的MNIST图片是一个784维的向量,我们重新把它还原为28x28维的图像。
  image_array = image_array.reshape(28, 28)
  # 保存文件的格式为 mnist_train_0.jpg, mnist_train_1.jpg, ... ,mnist_train_19.jpg
  filename = save_dir + 'mnist_train_%d.jpg' % i
  # 将image_array保存为图片
  # 先用scipy.misc.toimage转换为图像,再调用save直接保存。
  scipy.misc.toimage(image_array, cmin=0.0, cmax=1.0).save(filename)

print('Please check: %s ' % save_dir)

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

Python实现简单遗传算法(SGA)

Python实现简单遗传算法(SGA)

本文用Python3完整实现了简单遗传算法(SGA) Simple Genetic Alogrithm是模拟生物进化过程而提出的一种优化算法。SGA采用随机导向搜索全局最优解或者说近似...

使用python获取邮箱邮件的设置方法

首先我们要在邮箱的设置中开通那个POP3 然后我们要导入这些包 import poplib from datetime import datetime import jieba im...

python3 pathlib库Path类方法总结

这篇文章主要介绍了python3 pathlib库Path类方法总结,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下 1、调用库...

python生成n个元素的全组合方法

利用二进制反格雷码(bynary reflected Gray code)的方式生成n个元素的全组合,Cn1+Cn2+...+Cnn, 如在利用穷举方法解决背包问题时,就需要找出物品的所...

Python编程实现的简单神经网络算法示例

Python编程实现的简单神经网络算法示例

本文实例讲述了Python编程实现的简单神经网络算法。分享给大家供大家参考,具体如下: python实现二层神经网络 包括输入层和输出层 # -*- coding:utf-8 -*-...