Skip to content

Instantly share code, notes, and snippets.

@danielajisafe
Created July 15, 2018 15:22
Show Gist options
  • Save danielajisafe/b0d5b14f3bbae522bdfe110fa81555e7 to your computer and use it in GitHub Desktop.
Save danielajisafe/b0d5b14f3bbae522bdfe110fa81555e7 to your computer and use it in GitHub Desktop.
def HappyModel(input_shape):
"""
Implementation of the HappyModel.
Arguments:
input_shape -- shape of the images of the dataset
Returns:
model -- a Model() instance in Keras
"""
# Define the input placeholder as a tensor with shape input_shape. Think of this as your input image!
X_input = Input(input_shape)
# Zero-Padding: pads the border of X_input with zeroes
X = ZeroPadding2D((3, 3))(X_input)
# CONV -> BN -> RELU Block applied to X
X = Conv2D(32, (3, 3), strides = (1, 1), name = 'conv0')(X)
X = BatchNormalization(axis = 3, name = 'bn0')(X)
X = Activation('relu')(X)
# MAXPOOL
X = MaxPooling2D((2, 2), name='max_pool')(X)
# FLATTEN X (means convert it to a vector) + FULLYCONNECTED
X = Flatten()(X)
X = Dense(1, activation='sigmoid', name='fc')(X)
# Create model. This creates your Keras model instance, you'll use this instance to train/test the model.
model = Model(inputs = X_input, outputs = X, name='HappyModel')
return model
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment