keras-resnet
keras-resnet copied to clipboard
How can we add or change a layer
Please guide me on how can I change the in btw layer or last layer
shape, classes = (150, 150, 3), 11
x = keras.layers.Input(shape)
model = keras_resnet.models.ResNet50(x, classes=classes)
model.add(Dense(11, activation = "softmax")) <----- How can I add this layer? or make it last layer
model.compile(Adam(lr=0.0001), loss='binary_crossentropy', metrics=['accuracy'])
Thanks, Arindam
You can do as:
x = keras.layers.Input(shape)
resnet_arch= keras_resnet.models.ResNet50(x, classes=classes)
layer_1 = (keras.layers.Dense(11, activation='softmax', name="Dense_1"))(resnet_arch.output)
model = keras.models.Model(inputs=resnet_arch.input, outputs=layer_1)
model.compile(Adam(lr=0.0001), loss='binary_crossentropy', metrics=['accuracy'])
model.summary() ```