broadinstitute / keras-resnet

Keras package for deep residual networks
Other
300 stars 127 forks source link

How can we add or change a layer #70

Open Arindam-Jain opened 3 years ago

Arindam-Jain commented 3 years ago

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

ro-hit81 commented 2 years ago

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