keras-resnet icon indicating copy to clipboard operation
keras-resnet copied to clipboard

How can we add or change a layer

Open Arindam-Jain opened this issue 4 years ago • 1 comments

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

Arindam-Jain avatar Feb 24 '21 10:02 Arindam-Jain

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() ```

ro-hit81 avatar Jul 21 '22 17:07 ro-hit81