将mnist数据集保存为可见图片

最新推荐文章于 2021-01-11 15:18:17 发布

转载最新推荐文章于 2021-01-11 15:18:17 发布 · 408 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/dudu1992/p/8908178.html

文章标签：

#python #人工智能 #matlab

本文介绍如何使用TensorFlow下载MNIST数据集，并将数据集中的训练图像保存为图片文件。文中提供了完整的Python代码示例，展示了从下载数据到转换并保存图像的全过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#下载mnist数据集

from tensorflow.examples.tutorials.mnist import input_data
import tensorflow as tf
mnist = input_data.read_data_sets("/Users/dudu/Desktop/project/img",one_hot=True)

#保存为图片

import scipy.misc
import os
save_dir = '/Users/dudu/Desktop/project/img/raw'
if os.path.exists(save_dir) is False:
    os.makedirs(save_dir)
for i in range(20):
    image = mnist.train.images[i,:]
    image = image.reshape(28,28)
    file = save_dir+'mnist_train_%d.jpg' % i
    scipy.misc.toimage(image,cmin=0.0,cmax=1.0).save(file)

转载于:https://www.cnblogs.com/dudu1992/p/8908178.html