tensorflow.keras에서 기본으로 제공해주는 Datasets(mnist)예제을 통해 이미지을 확인해보고 차원수를 늘리거나 줄이는 예제을 해보겠다.

---

먼저 Numpy, matplotlib.pyplot, tensorflow 라이브러리을 불러오고 텐서플로우 케라스에서 기본으로 제공해주는 Datasets(MNIST)을 불러온다

import numpy as np
import matplotlib.pyplot as plt

import tensorflow as tf

%matplotlib inline

from tensorflow.keras import datasets

mnist = datasets.mnist

(train_x, train_y), (test_x, test_y) = mnist.load_data()

리스트 형태의 값으로 받아짐
train_x, text_x : 이미지
train_y, text_y : 해당 이미지 제목

# 차원수 확인
train_x.shape #(60000, 28, 28)

plt.imshow(train_x[0],'gray')
plt.title(test_y[0])
plt.colorbar()
plt.show()