인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

Inflearn Community Q&A

그접시제가살게요's profile image
그접시제가살게요

asked

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

Easily improve performance - Pretrained Model

모델 학습이 될 때, 입력이 궁금합니다.

Written on

·

222

1

이번 강의에서 train, test 데이터를 사용할 때, CIFAR10 이미지를 사용하는데요.

여기서 이미지 사이즈가 (3,32,32)가 되는데 resnet18 모델에 입력으로 넣어주어도 상관이 없나요?

resnet18 모델의 인풋이 (224,224)인거 같아서 달라도 문제가 없는 것인지 궁금합니다.

인공신경망pytorch딥러닝python머신러닝 배워볼래요?

Answer 1

0

dlbro님의 프로필 이미지
dlbro
Instructor

안녕하세요!

좋은 질문 감사합니다!!

CNN은 필터가 움직이면서 피쳐맵을 만들어가기 때문에 입력 이미지의 높이, 너비 상관없이 연산이 가능합니다. 다만, 이미지 사이즈가 차이가 큰 업무에 전이학습을 할 경우 성능 차이가 클 수 있습니다. 중요한 것은 채널 수는 같아야 합니다. 채널 수는 다르면 연산이 안 됩니다.

따라서 CIFAR10을 위해 만든 모델을 채널 수가 다른 MNIST와 같은 데이터 학습에 사용하실 때에는 첫번 째 합성곱 층의 채널 수를 변경해 주셔야만 합니다. (예: nn.Conv2d(3,16,3,1) -> nn.Conv2d(1,16,3,1))

 

열공하세요!

빠른 답변 감사드립니다!ㅎㅎ

그접시제가살게요's profile image
그접시제가살게요

asked

Ask a question