Commit b7f6ecb4 authored by sjjsmuel's avatar sjjsmuel

deeper FC Layers

parent 3e75a279
......@@ -32,13 +32,13 @@ class Resnet50(NetworkBase):
for layer in base_model.layers:
layer.trainable = False
x = GlobalAveragePooling2D()(base_model.output)
x = Dropout(0.5)(x)
#x = GlobalAveragePooling2D()(base_model.output)
x = Flatten()(base_model.output)
x = Dropout(0.3)(x)
x = Dense(128)(x)
x = Dropout(0.3)(x)
out = Dense(self.NUM_CLASSES, activation='softmax', name='probs')(x)
model = Model(base_model.input, out)
#for layer in model.layers[:-39]:
# layer.trainable = False
return model
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment