강의

멘토링

커뮤니티

인프런 커뮤니티 질문&답변

Jaewoo Choi님의 프로필 이미지
Jaewoo Choi

작성한 질문수

[파이토치] 실전 인공지능으로 이어지는 딥러닝 - 기초부터 논문 구현까지

이미지 분류 - 합성곱 신경망(CNN)

fc1 질문드립니다

작성

·

178

0

5 x 5 피처맵이 어디서 나온건가요...? 16은 전단계 output layer를 받는걸로 이해를 했는데 5 x 5가 어디서 어떤 원리로 나온건지 모르겠습니다 ...
좋은 강의 감사합니다

답변 1

3

딥러닝호형님의 프로필 이미지
딥러닝호형
지식공유자

너비 높이가 같은 입력 이미지를 사용할 경우 피쳐맵의 가로 세로 크기는 일반적으로 다음과 같이 계산 됩니다. 

output size = 1 + (input size + 2*padding - filter size)/stride

따라서 각 층을 위 식과 필터 개수를 같이 고려하여 계산하면 다음과 같은 결과를 얻게 됩니다.

입력 이미지 크기 3x32x32

conv1의 피쳐맵 크기 1+(32-5)/1=28 -> 6x28x28

pool1의 피쳐맵 크기 1+(28-2)/2=14 -> 6x14x14

conv2의 피쳐맵 크기 1+(14-5)/1=10 -> 16x10x10

pool2의 피쳐맵 크기 1+(10-2)/2=5 -> 16x5x5

Jaewoo Choi님의 프로필 이미지
Jaewoo Choi
질문자

감사합니다 제가 계산을 잘못했었내요ㅠㅠ처음에 ㅠㅠ

Jaewoo Choi님의 프로필 이미지
Jaewoo Choi

작성한 질문수

질문하기