Commit cb760532 authored by sjjsmuel's avatar sjjsmuel

resnet

parent da94ca5a
......@@ -36,7 +36,7 @@ class Resnet50(NetworkBase):
x = GlobalAveragePooling2D()(base_model.output)
x = Dropout(0.4)(x)
x = Dense(128, activation='relu', activity_regularizer=regularizers.l2(0.005))(x)
x = Dense(128, activity_regularizer=regularizers.l2(0.005))(x)
x = Dropout(0.2)(x)
out = Dense(self.NUM_CLASSES, activation='softmax', activity_regularizer=regularizers.l2(0.005), name='probs')(x)
......
......@@ -99,7 +99,7 @@ test_dataset = test_loader.load_dataset()
# Create Network
network = VGG_16(n_classes, img_width, img_height, channels, base_model_file)
network = Resnet50(n_classes, img_width, img_height, channels, base_model_file)
model = network.get_model()
#compile the 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