ML_DL/딥러닝 공부하기
[파이토치로 시작하는 딥러닝] Part-3 CNN
swwho
2023. 7. 2. 14:52
728x90
반응형
Convolution
- 이미지 위에서 stride 만큼 fillter를 이동하면서 겹쳐지는 부분의 각 원소의 값을 곱하여 모두 더한 값을 출력
- nn.Conv2d(1,1,1) : 입력채널 1 / 출력 채널 1 / 커널 크기 3x3
- input type : torch.Tensor / output type : N(batch size), C(channel), H(height), W(width)
- output size = (input size - fillter size + (2 * padding)) / stride + 1
- pooling : 이미지의 사이즈를 줄이거나, fillter 내의 평균값을 추출할 수 있다.