CNN - pytorch convnet에서 zero padding 값 구하는 공식

 

 

 

input이 7x7,

필터가 3x3인 경우

zero padding 값을 몇으로 해야 7x7의 결과를 얻을 수 있을까?

 

stride가 1인 경우 필터 크기를 2로 나눈 몫이 zero padding 개수가 된다.

그렇다면 stride가 2이거나 그 이상인 경우는 어떻게 계산해야 할까?

 

내 나름대로 식을 한번 구해봤다.

 

input 크기:  I(대문자 i) 

   ex) 7x7 -> 7

필터 크기: F

stride: S

원파는 output 크기: O

padding: P

 

공식: P = ( OS - I + F - 1 ) / 2

 

공식에 대입을 해보면,

P = ( 7*2 - 7 + 3 -1 ) / 2

  = 9 / 2

  = 4(몫)

 

확인을 해보자.

 

행만 확인해보면 7번의 스탭이 나오는걸 확인 할 수 있다.(열도 같음)

그렇게해서 우리가 원하는 7x7의 아웃풋이 나오는걸 예상 할 수 있다.