더북(TheBook)

Note ≡


GlobalAveragePooling은 플래튼(flatten)이 일어나기 직전 혹은 마지막 합성곱층에 적용하는 방식으로 각 특성 맵의 평균값을 뽑아 벡터로 만듭니다. 이해를 돕고자 마지막 합성곱층의 특성 맵 개수는 세 개이고 각각의 크기는 3×3이라고 가정해 봅시다.

▲ 그림 5-29 GlobalAveragePooling

이 예시에서 GlobalAveragePooling은 각 특성 맵에 대해 모든 값을 더하고, GlobalMaxPooling은 모든 값 중 최댓값을 뽑아 벡터로 만듭니다. 평균(average)을 취하면 1+2+1을 한 후 9로 나누어야 맞겠지만, GlobalAverage Pooling은 단순하게 합으로 대체합니다.

이제 사전 훈련된 합성곱층의 가중치를 고정하고 시그모이드 활성화 함수가 포함된 밀집층(완전연결층)을 추가하겠습니다.

▲ 그림 5-30 훈련된 계층을 고정하고 밀집층 추가

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.