딥러닝

딥러닝(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)

 

반응형