더북(TheBook)

네트워크의 각 부분들을 통과할 때마다 입력과 출력의 형태가 바뀌는데, 그 계산은 다음 수식과 같습니다(수식은 5장에 자세히 설명되어 있습니다).

 

Conv2d 계층에서의 출력 크기 구하는 공식

출력 크기 = (W-F+2P)/S+1

W: 입력 데이터의 크기(input_volume_size)

F: 커널 크기(kernel_size)

P: 패딩 크기(padding_size)

S: 스트라이드(strides)

 

MaxPool2d 계층에서의 출력 크기 구하는 공식

출력 크기 = IF/F

IF: 입력 필터의 크기(input_filter_size, 또한 바로 앞의 Conv2d의 출력 크기이기도 합니다)

F: 커널 크기(kernel_size)

각각의 계층에 대한 결과는 다음 그림과 같습니다.

▲ 그림 6-5 합성곱층과 풀링층의 크기(형태) 계산

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