본문 바로가기
Study/Machine Learning

[Tensorflow 2.0] dataset(MNIST)

by Jamie Lim 2021. 1. 11.

Data Process (MNIST)

 

데이터 불러오기

TensorFlow에서 제공해주는 데이터셋(MNIST) 불러오기

 

데이터 shape 확인하기

Image Dataset 들여다보기

불러온 데이터셋에서 이미지 데이터 하나만 뽑아 시각화하여 확인하기

 

- 데이터 하나 뽑기

 

- 시각화해서 확인하기

 

 


Channel 관련

[Batch Size, Height, Width, Channel]
GrayScale이면 1, RGB이면 3으로 만들어줘야함

 

데이터 차원수 늘리기 (numpy)

TensorFlow 패키지를 불러와 데이터 차원수 늘리기 (tensorflow)

* matplotlib로 이미지 시각화 할 때는 gray scale의 이미지는 3번쨰 dimension이 없으므로, 2개의 dimension으로 gray scale로 차원 조절해서 넣어줘야함

시각화하기

 


Label Dataset 들여다보기

Label 하나를 열어서 Image와 비교하여 제대로 들어갔는지. 어떤 식으로 저장 되어있는지 확인

label 하나 뽑아보기 시각화하기

 


OneHot Encoding

컴퓨터가 이해할 수 있는 형태로 변환해서 Label을 주도록 함

 

tensorflow.keras.utils.to_categorical

 

label 확인하여 to_categorical을 사용한다

 

onehot encoding으로 바꾼 것과 이미지 확인하기

댓글