딥러닝
딥러닝(9)Flatten 라이브러리 없이, 이미지를 평탄화 하는 방법
개발연습자1
2022. 12. 29. 18:01
지금까지 이미지 데이터를 평탄화 하기 위해선 Flatten 라이브러리를 썼다. 사실 Flatten 라이브러리는 텐서플로우가 버전이 업 된 다음부터 나온거라 그전에는 평탄화 작업을 Flatten 라이브러리 없이 썼다.
지금부터 그 방법을 알아보자
데이터를 불러오는것까진 Flatten 라이브러리 글과 동일하다.
# 0~255의 데이터이므로 255로 나눈다.
X_train=X_train/255.0
X_test = X_test/255.0
이렇게 나누게 되면 저절로 데이터가 0~1사이의 값으로 바뀐다.
이상태에서 데이터를 리셰이프 해준다.
#모양을 행수와 컬럼과 컬럼안에 데이터수를 곱한 값으로 써준다.
X_train=X_train.reshape(60000,28*28)
X_test=X_test.reshape(10000,28*28)
그럼 X데이터가 평탄화 되어 있는것을 알수 있다.
X_train.shape
>> (60000,784)
X_test.shape
>> (10000,784)
반응형