Deep Learning/밑바닥부터 시작하는 딥러닝

DL - #7, Pooling 계층 구현

KimTory 2022. 2. 28. 23:46

풀링 계층 구현하기

 풀링 계층 구현도 합성곱 계층과 마찬가지로 im2col을 사용해 입력 데이터를 전개합니다.

 단, 풀링의 경우엔 채널 쪽이 독립적이라는 점이 합성곱 계층 때와 다릅니다.

 구체적으로는 아래 그림과 같이 풀링 적용 영역을 채널마다 독립적으로 계산합니다.

 

입력 데이터에 풀링 적용 영역을 전개 (2 X 2 풀링의 예)

 

일단 이렇게 전개한 후, 전개한 행렬에서 행별 최댓값을 구하고 적절한 형상으로 성형하기만 하면 됩니다.

 

풀링 계층 구현의 흐름 : 풀링 적용 영역에서 가장 큰 원소는 회색으로 표시

 

 이상이 풀링 계층의 forward 처리 흐름입니다.

 

 

 풀링 계층 구현은 아래와 같이 세 단계로 진행합니다.

 

1. 입력 데이터를 전개한다.

2. 행별 최댓값을 구한다.

3. 적절한 모양으로 성형한다.

'Deep Learning > 밑바닥부터 시작하는 딥러닝' 카테고리의 다른 글

#7, CNN 시각화 작업  (0) 2022.02.28
#7, CNN 구현  (0) 2022.02.28
DL - #7, CNN 계층 구현  (0) 2022.02.28
DL - #6, 적절한 하이퍼 파라미터 설정  (0) 2022.02.15
DL - #6, Batch Normalization  (0) 2022.02.14