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'))

コメント

このブログの人気の投稿

slackでgeneralの投稿を全削除する

Python SQLite スレッド間でコネクションの使いまわしは出来ない

slackで投稿内容を自動翻訳する(3/5)slackにおけるメッセージの構造