강의

멘토링

로드맵

Inflearn Community Q&A

6zww5fsjcp2449's profile image
6zww5fsjcp2449

asked

[PyTorch] Deep Learning Leading to Practical AI - From Basics to Paper Implementation

Image Classification - Convolutional Neural Network (CNN)

fc1 질문드립니다

Written on

·

174

0

5 x 5 피처맵이 어디서 나온건가요...? 16은 전단계 output layer를 받는걸로 이해를 했는데 5 x 5가 어디서 어떤 원리로 나온건지 모르겠습니다 ...
좋은 강의 감사합니다
머신러닝 배워볼래요? pytorch딥러닝python인공신경망

Answer 1

3

dlbro님의 프로필 이미지
dlbro
Instructor

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

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
Questioner

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

6zww5fsjcp2449's profile image
6zww5fsjcp2449

asked

Ask a question