Commit 24423894 authored by sjjsmuel's avatar sjjsmuel

network modification

parent 09b9fbad
......@@ -32,11 +32,12 @@ class VGG_16(NetworkBase):
base_model.save('input/vgg_base_model.h5')
x = GlobalAveragePooling2D()(base_model.output)
x = Dense(4096, activation='relu')(x)
x = Dense(4096, activation='relu')(x)
x = Dense(256, activation='relu')(x)
x = Dense(256, activation='relu')(x)
out = Dense(self.NUM_CLASSES, activation='softmax', name='probs')(x)
model = Model(base_model.input, out)
print(model.summary())
for layer in model.layers[:15]:
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