Symfoware

Python mnistデータセットの内容を画像ファイルに出力する(pillow使用)

機械学習のサンプルでよく使用する手書き数字のminstデータセット。
こんな感じで利用していると思います。

from tensorflow.keras.datasets import mnist
(x_train, y_train), (x_test, y_test) = mnist.load_data()

x_train、x_testに画像データの配列。
y_train、y_testに画像データの値が取得できるのですが、実際の内容を確認したい。

np.arryを画像ファイルに出力

こちらを参考にさせていただきました。
numpyの2次元配列をpillowで保存するサンプル

x_train配列の最初のデータを画像ファイルとして出力してみます。

import numpy as np
from PIL import Image
from tensorflow.keras.datasets import mnist
(x_train, y_train), (x_test, y_test) = mnist.load_data()
# 画像データ
# print(x_train[0])
# ラベルの値
print(y_train[0])
img = Image.fromarray(np.uint8(x_train[0]))
img.save('x_train.png')

実行すると「5」と出力されるので、画像は数字の5が記載されているはずです。
出力されたx_train.pngはこちら。



ちゃんと出力できました。




【参考URL】

numpyの2次元配列をpillowで保存するサンプル

関連記事

• Python mnistデータセットの画像をtkinterで表示する
• Ubuntu 20.04 + Python 3で「ModuleNotFoundError: No module named 'PIL.ImageTk'」エラー
• Python mnistデータセットの内容を画像ファイルに出力する(pillow使用)
• Python websocket-clientでバイナリデータの送受信
• Python websocket-clientでBottleフレームワークと通信する