inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

Cifar10을 vgg16으로 전이학습에서 input size 대한 질문

375

김주호

작성한 질문수 1

0

안녕하세요. 전이학습을 통해 Cifar10 dataset을 Vgg16 모델에서 미리 훈련된 ImageNet 가중치를 로드하여 학습을 시키려고 하는데,

아래와 같이 모델을 구성하게 되면 훈련을 진행할 수는 있지만 전이학습의 이점을(빠른 학습 속도) 얻는 것이 어렵지 않나 생각이 되어 질문드립니다.

또한 의문이 드는 것은

VGG16의 원래 모델에서의 input size는 224 x 244 x 3로 알고 있는데

Cifar10의 image_size는 32 x 32 x 3 이라 너무 이미지가 작아서 잘 학습이 안되는 것 같다는 생각이 듭니다.

혹시 Cifar10의 이미지를 upsampling 하는 방법이나 accuracy를 높일 수 있는 다른 방법이 있을까요?

 

from tensorflow.keras.optimizers import RMSprop

conv_base = VGG16(weights = 'imagenet', input_shape=(32, 32, 3), include_top=False)
def build_model_with_pretrained(conv_base): model = Sequential() model.add(conv_base) model.add(Flatten()) model.add(Dense(256, activation= 'relu')) model.add(Dense(1, activation= 'sigmoid')) model.compile(loss = 'binary_crossentropy', optimizer = RMSprop(learning_rate = 2e-5),
metrics = ['accuracy']) return model

model = build_model_with_pretrained(conv_base) model.summary()

vgg16 transferlearning resize upsampling cifar10

답변 0

딥러닝 학습 중, Training loss와 Test accuracy 간의 연관성

0

894

0

이미지데이터 학습 후 평가 시 그래프가 튀는 경우와 정확도 산정

0

817

0

data pipeline 관련 질문입니다.

0

261

3

resize에 관해서 질문입니다.

0

225

1

CNN Cifar10 VGG16으로 전이학습 시 val_accuracy가 0.1로 고정되어 나옵니다

0

1054

6

keras vgg16 pretrained 모델 사용할 때 입력 shape에 대해서 질문이 있습니다

0

487

2