Skip to content

Instantly share code, notes, and snippets.

@amankharwal

amankharwal/flower.py Secret

Created November 24, 2020 08:34
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save amankharwal/6380923a4f2fd68fb64dac00a0aaed8c to your computer and use it in GitHub Desktop.
Save amankharwal/6380923a4f2fd68fb64dac00a0aaed8c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
flower.py
model = Sequential()
model.add(Conv2D(filters = 64, kernel_size = (3,3),padding = 'Same',activation ='relu', input_shape = (SIZE,SIZE,3)))
model.add(MaxPooling2D(pool_size=(2,2)))
model.add(Conv2D(filters = 128, kernel_size = (3,3),padding = 'Same',activation ='relu'))
model.add(Conv2D(filters = 128, kernel_size = (3,3),padding = 'Same',activation ='relu'))
model.add(Conv2D(filters = 128, kernel_size = (3,3),padding = 'Same',activation ='relu'))
model.add(MaxPooling2D(pool_size=(2,2)))
model.add(Flatten())
model.add(Dense(128, activation='relu'))
model.add(Dense(64, activation='relu'))
model.add(Dropout(rate=0.5))
model.add(Dense(5, activation = "softmax"))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment