군침이 싹 도는 코딩
Flatten라이브러리 없이 이미지를 평탄화 하는법 본문
이미지를 인공지능 학습 시킬때 인풋쉐입으로 사용하기 위해서는
Flatten 이라는 라이브러리를 사용한다
사용법과 정의는 이 글 참고 : https://mugoori.tistory.com/132
Flatten 라이브러리 사용 이유
def build_model(): model = Sequential() model.add( Flatten()) model.add( Dense(128,'relu') ) model.add( Dense(64,'relu') ) model.add( Dense(10, 'softmax')) model.compile('adam','sparse_categorical_crossentropy',['accuracy']) return model # 위 코드는
mugoori.tistory.com
X_train.shape
>>> (60000, 28, 28)
X_test.shape
>>> (10000, 28, 28)# 이러한 이미지 데이터를 Flatten 없이 처리 할수도 있다
X_train = X_train.reshape(60000,784)
X_test = X_test.reshape(10000,784)# 먼저 28행 28열인것을 곱해준다
그 후 리쉐이브를 사용해서 사진의 갯수, 행*열 한 값을 써주면된다
X_train.shape
>>> (60000, 784)
X_test.shape
>>> (10000, 784)# 결과를 보니 2차원으로 잘 변형이 되었다
'Python > Deep Learning' 카테고리의 다른 글
| 텐서플로우의 모델 save / load 하는 법 (0) | 2022.12.29 |
|---|---|
| epoch 중에 벨리데이션 데이터를 처리하는 방법 중 validation_data 파라미터 사용법 (0) | 2022.12.29 |
| epoch 에 따른 데이터의 overfitting 과 해결법 (callback class) (0) | 2022.12.29 |
| softmax 결과를 레이블 인코딩으로 바꾸는 법 (argmax) (0) | 2022.12.29 |
| activation 함수 softmax / 분류 문제 loss 셋팅 (0) | 2022.12.28 |
'Python/Deep Learning' Related Articles
more
