MNIST をpngで出力
通番とラベルをファイル名にしてpng形式で出力
from PIL import Image import os path = "MNIST_data/" from tensorflow.examples.tutorials.mnist import input_data mnist = input_data.read_data_sets(path, one_hot=False) batch = mnist.train.next_batch(mnist.train.num_examples) for i in range(mnist.train.num_examples): data = batch[0][i] label = batch[1][i] im = Image.new('L', (28,28)) im.putdata(data, 256) im.save(os.path.join(path, 'train_{0:05d}'.format(i) + '_{0:01d}'.format(label) + '.png')) batch = mnist.test.next_batch(mnist.test.num_examples) for i in range(mnist.test.num_examples): data = batch[0][i] label = batch[1][i] im = Image.new('L', (28, 28)) im.putdata(data, 256) im.save(os.path.join(path, 'test_{0:05d}'.format(i) + '_{0:01d}'.format(label) + '.png'))
コメント
コメントを投稿