Data Preprocess (데이터 전처리)

import numpy as np
import matplotlib.pyplot as plt
import tensorflow as tf

%matplotlib inline

데이터 불러오기

• MNIST 데이터셋은 TensorFlow에서 제공해줍니다.

from tensorflow.keras import datasets

• 데이터 shape 확인해보기

mnist = datasets.mnist
(train_x, train_y), (test_x,text_y) = mnist.load_data()
train_x.shape

Image Dataset 조회하기

불러온 데이터셋을 가지고 이미지를 시각화시켜서 조회해보겠습니다.

• 데이터 하나를 추출해서 시각화

image = train_x[0] #범위 내의 아무 숫자나 넣으셔도 상관없어요
image.shape #28*28 size
plt.imshow(image,'gray')
plt.show()

Channel

[Batch Size, Height, Channel] 순입니다.

GreyScale이면 1, RGB면 3으로 설정을 해줘야합니다.