앞서 생성했던 모델(model)에 밀집층을 추가하는 코드는 다음과 같습니다.
코드 5-15 ResNet50 네트워크에 밀집층 추가
model.trainable = False
model = Sequential([model,
Dense(2, activation='sigmoid')]) ------ 시그모이드 함수가 포함된 밀집층 추가
model.summary()
다음은 밀집층이 추가된 ResNet50 네트워크의 출력 결과입니다.
Model: "sequential_1"
________________________________________________________________
Layer (type) Output Shape Param #
================================================================
resnet50 (Functional) (None, 1000) 25636712
________________________________________________________________
dense_1 (Dense) (None, 2) 2002
================================================================
Total params: 25,638,714
Trainable params: 2,002
Non-trainable params: 25,636,712
________________________________________________________________