pytorch

CNN 모델 중 정말 조상님이라 할 수 있는 LeNet의 버전 중 LeNet-5를 직접 코드로 구현해보려고 한다. LeNet-5 구조 저 숫자가 어떻게 나온거지? 저기 있는 사이즈, 채널에 대해 숫자가 어떻게 계산되어 나왔는지 정리를 해보았다. 각 레이어들 정리 Input size : 32X32 1채널 C1 : 28 * 28 커널 수 6개 S2 : 14*14 커널 수 6개 C3 : 10*10 커널 수 16개 S4 : 5*5 커널 수 16개 C5 : 120 F6 : 84 OUPUT : 10 계산을 위한 공식 H : 입력 데이터의 Height P : Padding사이즈 FH : Filter(Kernal)의 Height FW : Filter(Kernal)의 Width S : Stride 크기 Convoluti..
Berno
'pytorch' 태그의 글 목록