이제 앞서 우리가 만든 모델에 이미지넷 데이터셋에서 미리 학습된 모델인 VGGNet을 가지고 오는 예제를 실행해 보겠습니다. VGGNet은 옥스포드 대학의 연구 팀 VGG에 의해 개발된 모델로, 2014년 이미지넷 이미지 인식 대회에서 2위를 차지한 모델입니다. 학습 구조에 따라 VGG16, VGG19 등 이름이 주어졌는데, 우리는 VGG16을 사용하겠습니다.

TIP

다음은 모델 이름을 transfer_model로 정하고 VGG16을 불러온 모습입니다. include_top은 전체 VGG16의 마지막 층, 즉 분류를 담당하는 곳을 불러올지 말지를 정하는 옵션입니다. 우리가 만든 로컬 네트워크를 연결할 것이므로 False로 설정합니다. 또한, 불러올 부분은 새롭게 학습되는 것이 아니므로 학습되지 않도록 transfer_model.trainable 옵션 역시 False로 설정합니다.

transfer_model = VGG16(weights='imagenet', include_top=False, input_shape=(150,150,3))
transfer_model.trainable = False
transfer_model.summary()