Tensorflow

tensorflow keras datasets (MNIST)

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

댓글 남기기

이메일은 공개되지 않습니다. 필수 입력창은 * 로 표시되어 있습니다