Skip to content

Instantly share code, notes, and snippets.

@nitin-bommi
Last active July 15, 2020 11:47
Show Gist options
  • Save nitin-bommi/974e5067504a6451245b14ee2139aeb5 to your computer and use it in GitHub Desktop.
Save nitin-bommi/974e5067504a6451245b14ee2139aeb5 to your computer and use it in GitHub Desktop.
model = tf.keras.models.Sequential([
tf.keras.layers.Conv2D(32, (3,3), activation='relu', input_shape=(150, 150, 3)),
tf.keras.layers.MaxPooling2D(2, 2),
tf.keras.layers.Conv2D(64, (3,3), activation='relu'),
tf.keras.layers.MaxPooling2D(2,2),
tf.keras.layers.Conv2D(128, (3,3), activation='relu'),
tf.keras.layers.MaxPooling2D(2,2),
tf.keras.layers.Conv2D(128, (3,3), activation='relu'),
tf.keras.layers.MaxPooling2D(2,2),
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(512, activation='relu'),
tf.keras.layers.Dense(1, activation='sigmoid')
])
model.summary()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment